diff --git a/pybop/problems/base_problem.py b/pybop/problems/base_problem.py
index 90fc67f92..2063b3111 100644
--- a/pybop/problems/base_problem.py
+++ b/pybop/problems/base_problem.py
@@ -1,3 +1,4 @@
+import pybop
 import numpy as np