diff --git a/hol/1_1_DEMO_basic_data_exploration_transformation.ipynb b/hol/1_1_DEMO_basic_data_exploration_transformation.ipynb
index bd7859f..b1be351 100644
--- a/hol/1_1_DEMO_basic_data_exploration_transformation.ipynb
+++ b/hol/1_1_DEMO_basic_data_exploration_transformation.ipynb
@@ -1258,7 +1258,7 @@
     }
    ],
    "source": [
-    "# Get all categorical columns\n",
+    "# Get all numerical columns\n",
     "numeric_types = [T.DecimalType, T.LongType, T.DoubleType, T.FloatType, T.IntegerType]\n",
     "numeric_columns = [c.name for c in snowpark_df.schema.fields if type(c.datatype) in numeric_types]\n",
     "numeric_columns"
@@ -1289,7 +1289,7 @@
     }
    ],
    "source": [
-    "# Get all numerical columns\n",
+    "# Get all categorical columns\n",
     "categorical_types = [T.StringType]\n",
     "categorical_columns = [c.name for c in snowpark_df.schema.fields if type(c.datatype) in categorical_types]\n",
     "categorical_columns"