diff --git a/demo_notebooks/demo_multiday.ipynb b/demo_notebooks/demo_multiday.ipynb
index 3d370d9..10df799 100644
--- a/demo_notebooks/demo_multiday.ipynb
+++ b/demo_notebooks/demo_multiday.ipynb
@@ -16,7 +16,7 @@
     "\n",
     "### Main files\n",
     "\n",
-    "The following files are used for the simulation.\n",
+    "The following modules are used for the simulation.\n",
     "\n",
     "- ``simulation.py``: This script stores information on the ``Simulation`` class and subclasses. These classes are used to set up and solve the optimisation problem. For the multiday problem, we load the ``MultidayMIP`` class.\n",
     "- ``results.py``: The ``Results`` class stores the outputs of the simulation (optimised schedules, runtimes, and objective functions), and handles the visualisation and simulation statistics.\n",
@@ -44,13 +44,11 @@
     "import joblib\n",
     "import json\n",
     "import pandas as pd\n",
+    "import oasisabm as oa\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 MultidayMIP\n"
+    "from oa.input_data import data_reader\n",
+    "from oa.error_terms import GaussianError, EVError\n",
+    "from oa.simulation import MultidayMIP\n"
    ]
   },
   {