diff --git a/end_to_end/music_recommendation/03_train_model_lineage_registry_debugger.ipynb b/end_to_end/music_recommendation/03_train_model_lineage_registry_debugger.ipynb
index b35d2cf134..713761b922 100644
--- a/end_to_end/music_recommendation/03_train_model_lineage_registry_debugger.ipynb
+++ b/end_to_end/music_recommendation/03_train_model_lineage_registry_debugger.ipynb
@@ -491,9 +491,9 @@
    "source": [
     "%%time\n",
     "\n",
-    "train = True # True if training a new model, False if wanting to use an existing estimator once you've already trained\n",
+    "train_model = True # True if training a new model, False if wanting to use an existing estimator once you've already trained\n",
     "\n",
-    "if train:\n",
+    "if train_model:\n",
     "    print('Training the model')\n",
     "    xgb_estimator.fit(inputs = {'train': train_input, 'validation': val_input})\n",
     "    s3_debugger_output_path = xgb_estimator.latest_job_debugger_artifacts_path()\n",