diff --git a/HW2.ipynb b/HW2.ipynb
index 232adec..6801a51 100644
--- a/HW2.ipynb
+++ b/HW2.ipynb
@@ -1365,7 +1365,7 @@
       "    y, X = prepare_features(frame2008, featureslist)\n",
       "    clf = LogisticRegression()\n",
       "    parameters = {\"C\": np.logspace(-4, 3, num=num_p)}\n",
-      "    gs = GridSearchCV(clf, param_grid=parameters, cv=10)\n",
+      "    gs = GridSearchCV(clf, param_grid=parameters, cv=n_folds)\n",
       "    gs.fit(X, y)\n",
       "    return gs.best_params_, gs.best_score_\n"
      ],