diff --git a/src/pybamm_cookiecutter/__init__.py b/src/pybamm_cookiecutter/__init__.py
index 53e47d1..678b215 100644
--- a/src/pybamm_cookiecutter/__init__.py
+++ b/src/pybamm_cookiecutter/__init__.py
@@ -11,4 +11,5 @@
 
 __all__ : list[str] = [
     "__version__",
+    "pybamm",
 ]