diff --git a/demo_notebooks/demo.ipynb b/demo_notebooks/demo.ipynb
index 16cf86b..55e6f10 100644
--- a/demo_notebooks/demo.ipynb
+++ b/demo_notebooks/demo.ipynb
@@ -43,13 +43,12 @@
     "import joblib\n",
     "import json\n",
     "import pandas as pd\n",
+    "import oasisabm as as\n",
     "\n",
-    "import sys \n",
-    "sys.path.append('simulation/')\n",
-    "\n",
-    "from input_data import data_reader\n",
-    "from error_terms import GaussianError, EVError\n",
-    "from simulation import MIP\n"
+
+    "from oa.input_data import data_reader\n",
+    "from oa.error_terms import GaussianError, EVError\n",
+    "from oa.simulation import MIP\n"
    ]
   },
   {