diff --git a/01-g-h-filter.ipynb b/01-g-h-filter.ipynb
index 3ffb9d8..f01abf9 100644
--- a/01-g-h-filter.ipynb
+++ b/01-g-h-filter.ipynb
@@ -931,6 +931,7 @@
    },
    "outputs": [],
    "source": [
+    "from kf_book.gh_internal import plot_g_h_results\n",
     "def g_h_filter(data, x0, dx, g, h, dt):\n",
     "    pass #  your solution here\n",
     "\n",
@@ -966,7 +967,6 @@
     }
    ],
    "source": [
-    "from kf_book.gh_internal import plot_g_h_results\n",
     "import matplotlib.pylab as pylab\n",
     "\n",
     "def g_h_filter(data, x0, dx, g, h, dt=1.):\n",