diff --git a/mnist-classifaction/train_model.ipynb b/mnist-classifaction/train_model.ipynb
index 95060f4..3c79c26 100644
--- a/mnist-classifaction/train_model.ipynb
+++ b/mnist-classifaction/train_model.ipynb
@@ -91,7 +91,7 @@
       "source": [
        "import mlfoundry\n",
        "\n",
-       "client = mlfoundry.get_client()"
+       "client = mlfoundry.get_client(disable_analytics=True)"
       ]
      },
      {
@@ -163,7 +163,7 @@
       "outputs": [],
       "source": [
        "client.create_ml_repo(ML_REPO_NAME)\n",
-       "run = client.create_run(ml_repo=ML_REPO_NAME, name=\"train-model\")\n",
+       "run = client.create_run(ml_repo=ML_REPO_NAME, run_name=\"train-model\")\n",
        "\n",
        "# Plot some sample images\n",
        "plt.figure(figsize=(10, 5))\n",
@@ -350,4 +350,4 @@
     "nbformat": 4,
     "nbformat_minor": 4
    }
-   
\ No newline at end of file
+