diff --git a/.gitignore b/.gitignore
index 0ae5fd14c..3897edddc 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 *MetaDataFullWithYAMLContentCol.csv
-.vscode
\ No newline at end of file
+.vscode
+YAML_tester_USER.ipynb
\ No newline at end of file
diff --git a/YAML_tester.ipynb b/YAML_tester.ipynb
index b38c46fd4..2a2b752ba 100644
--- a/YAML_tester.ipynb
+++ b/YAML_tester.ipynb
@@ -11,7 +11,7 @@
     "import random\n",
     "import os\n",
     "\n",
-    "from animalai.animalai.envs.environment import AnimalAIEnvironment\n",
+    "from animalai.envs.environment import AnimalAIEnvironment\n",
     "from mlagents_envs.exception import UnityCommunicationException"
    ]
   },