diff --git a/main.ipynb b/main.ipynb
index eec174cd83da902ae5c5a52306af435b448bbaec..09e57a519470de980113356a73a18fc3d48f91d5 100755
--- a/main.ipynb
+++ b/main.ipynb
@@ -25,25 +25,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 2,
    "metadata": {},
-   "outputs": [
-    {
-     "ename": "ImportError",
-     "evalue": "attempted relative import with no known parent package",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mImportError\u001b[0m                               Traceback (most recent call last)",
-      "\u001b[1;32m/home/mullemat/code/sicom_image_analysis_project/src/methods/baseline/main.ipynb Cell 3\u001b[0m line \u001b[0;36m3\n\u001b[1;32m      <a href='vscode-notebook-cell:/home/mullemat/code/sicom_image_analysis_project/src/methods/baseline/main.ipynb#W2sZmlsZQ%3D%3D?line=0'>1</a>\u001b[0m \u001b[39mimport\u001b[39;00m \u001b[39mmatplotlib\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mpyplot\u001b[39;00m \u001b[39mas\u001b[39;00m \u001b[39mplt\u001b[39;00m\n\u001b[0;32m----> <a href='vscode-notebook-cell:/home/mullemat/code/sicom_image_analysis_project/src/methods/baseline/main.ipynb#W2sZmlsZQ%3D%3D?line=2'>3</a>\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39m.\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mutils\u001b[39;00m \u001b[39mimport\u001b[39;00m load_image, save_image, psnr, ssim\n\u001b[1;32m      <a href='vscode-notebook-cell:/home/mullemat/code/sicom_image_analysis_project/src/methods/baseline/main.ipynb#W2sZmlsZQ%3D%3D?line=3'>4</a>\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39msrc\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mforward_model\u001b[39;00m \u001b[39mimport\u001b[39;00m CFA\n\u001b[1;32m      <a href='vscode-notebook-cell:/home/mullemat/code/sicom_image_analysis_project/src/methods/baseline/main.ipynb#W2sZmlsZQ%3D%3D?line=4'>5</a>\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39msrc\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mmethods\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mbaseline\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mreconstruct\u001b[39;00m \u001b[39mimport\u001b[39;00m run_reconstruction\n",
-      "\u001b[0;31mImportError\u001b[0m: attempted relative import with no known parent package"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "import matplotlib.pyplot as plt\n",
     "\n",
-    "from ..utils import load_image, save_image, psnr, ssim\n",
+    "from src.utils import load_image, save_image, psnr, ssim\n",
     "from src.forward_model import CFA\n",
     "from src.methods.baseline.reconstruct import run_reconstruction"
    ]