diff --git a/src/analysis.ipynb b/src/analysis.ipynb
index 1863db0..8ce71f1 100644
--- a/src/analysis.ipynb
+++ b/src/analysis.ipynb
@@ -35,7 +35,7 @@
     "feat = \"concatenated_fv\"\n",
     "control = \"session\"\n",
     "\n",
-    "df_group = df.groupby([control])[\"rotation\",feat]\n",
+    "df_group = df.groupby([control])[[\"rotation\",feat]]\n",
     "\n",
     "g_id = list(df_group.groups.keys())"
    ]