diff --git a/notebooks/01_basics.ipynb b/notebooks/01_basics.ipynb
index da71a00..10f510e 100644
--- a/notebooks/01_basics.ipynb
+++ b/notebooks/01_basics.ipynb
@@ -24,6 +24,7 @@
     "# %% Import tensorflow and pyplot\n",
     "import tensorflow as tf\n",
     "import matplotlib.pyplot as plt"
+    "%matplotlib inline"
    ]
   },
   {