diff --git a/demo.ipynb b/demo.ipynb
index a2d1d8f..dc2146d 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -96,8 +96,8 @@
     "Try running the following cell to see if the installation worked.\n",
     "Please note that this may take up to a minute; if you see `[*]` next to the cell, that means it is still running.\n",
     "\n",
-    "If you get an output along the lines of     \n",
-    "```<module 'fpcup' from 'file/location/fpcup/fpcup/__init__.py'```    \n",
+    "If you get an output like    \n",
+    "```Succesfully imported fpcup version x.y.z.```    \n",
     "then the installation was successful!"
    ]
   },
@@ -109,7 +109,7 @@
    "outputs": [],
    "source": [
     "import fpcup\n",
-    "fpcup"
+    "fpcup.test()"
    ]
   },
   {
diff --git a/fpcup/__init__.py b/fpcup/__init__.py
index cd01ee2..80f08ab 100644
--- a/fpcup/__init__.py
+++ b/fpcup/__init__.py
@@ -4,3 +4,12 @@
 from .nn import dataset, network
 from .settings import DEFAULT_DATA, DEFAULT_OUTPUT, DEFAULT_RESULTS
 from .tools import RUNNING_IN_IPYTHON
+
+
+def test() -> None:
+    """
+    Simple function to assert the import was successful.
+    """
+    from importlib.metadata import version
+    module_version = version(__package__)
+    print(f"Succesfully imported {__package__} version {module_version}.")