diff --git a/predict-credit-churn/CreditChurn.ipynb b/predict-credit-churn/CreditChurn.ipynb
index d2ed94d..121b21c 100644
--- a/predict-credit-churn/CreditChurn.ipynb
+++ b/predict-credit-churn/CreditChurn.ipynb
@@ -225,7 +225,9 @@
     "    imputer = SimpleImputer(impute_strategy=\"most_frequent\", missing_values=\"Unknown\")\n",
     "    X = imputer.fit_transform(X, y)\n",
     "    \n",
-    "    return X"
+    "    return X\n",
+    "\n",
+    "X = preprocessing(X, y)"
    ]
   },
   {
@@ -243,7 +245,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "X = preprocessing(X, y)\n",
     "X_train, X_test, y_train, y_test = evalml.preprocessing.split_data(X, y, problem_type='binary',\n",
     "                                                                         test_size=.2)"
    ]