diff --git a/simpegEM/FDEM/__init__.py b/simpegEM/FDEM/__init__.py
index 562b9218..0b881e39 100644
--- a/simpegEM/FDEM/__init__.py
+++ b/simpegEM/FDEM/__init__.py
@@ -1,2 +1,2 @@
 from SurveyFDEM import *
-from FDEM import ProblemFDEM_e, ProblemFDEM_b, ProblemFDEM_j, ProblemFDEM_h
+from FDEM import BaseFDEMProblem, ProblemFDEM_e, ProblemFDEM_b, ProblemFDEM_j, ProblemFDEM_h, omega