diff --git a/psets/pset2.ipynb b/psets/pset2.ipynb
index 1213dc7..ab58b8d 100644
--- a/psets/pset2.ipynb
+++ b/psets/pset2.ipynb
@@ -253,7 +253,7 @@
     "\n",
     "# construct your A and b from M=8 and find β\n",
     "M = 8\n",
-    "A, b = autoregressive_model(training_data[1:end-100], M)\n",
+    "A, b = autoregressive_model(training_data, M)\n",
     "β = A \\ b # least-square fit"
    ]
   },