diff --git a/notebooks/aws/aws.ipynb b/notebooks/aws/aws.ipynb
index 26fc8a5..ce14f53 100644
--- a/notebooks/aws/aws.ipynb
+++ b/notebooks/aws/aws.ipynb
@@ -50,10 +50,10 @@
    "outputs": [],
    "source": [
     "%%stackql\n",
-    "SELECT region, instanceType, COUNT(*) as num_instances\n",
+    "SELECT region, instance_type, COUNT(*) as num_instances\n",
     "FROM aws.ec2.instances\n",
     "WHERE region = '$region'\n",
-    "GROUP BY instanceType"
+    "GROUP BY instance_type"
    ]
   },
   {
@@ -62,7 +62,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "_.plot(kind='pie', y='num_instances', labels=_['instanceType'], title='Instances by Type', autopct='%1.1f%%')"
+    "_.plot(kind='pie', y='num_instances', labels=_['instance_type'], title='Instances by Type', autopct='%1.1f%%')"
    ]
   },
   {
@@ -90,16 +90,6 @@
     "WHERE region = '$region'"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "%%stackql\n",
-    "select PolicyName, Arn from aws.iam.policies where region = '$region'"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -108,7 +98,7 @@
    "source": [
     "%%stackql\n",
     "SELECT region, function_name\n",
-    "FROM aws.lambda.functions\n",
+    "FROM aws.lambda.functions_list_only\n",
     "WHERE region IN (\n",
     "'us-east-1',\n",
     "'us-east-2',\n",