diff --git a/simpegEM/TDEM/BaseTDEM.py b/simpegEM/TDEM/BaseTDEM.py
index 72a6e6fa..d3b648da 100644
--- a/simpegEM/TDEM/BaseTDEM.py
+++ b/simpegEM/TDEM/BaseTDEM.py
@@ -1,6 +1,7 @@
 from SimPEG import Solver
 from SimPEG.Problem import BaseTimeProblem
 from simpegEM.Utils import Sources
+from SurveyTDEM import FieldsTDEM
 from scipy.constants import mu_0
 from SimPEG.Utils import sdiag, mkvc
 from SimPEG import Utils, Mesh
diff --git a/simpegEM/TDEM/TDEM_b.py b/simpegEM/TDEM/TDEM_b.py
index df05f9e6..e41c5644 100644
--- a/simpegEM/TDEM/TDEM_b.py
+++ b/simpegEM/TDEM/TDEM_b.py
@@ -1,7 +1,7 @@
 from BaseTDEM import ProblemBaseTDEM
 from SimPEG.Utils import mkvc
 import numpy as np
-from SurveyTDEM import SurveyTDEM1D
+from SurveyTDEM import SurveyTDEM1D, FieldsTDEM
 
 class ProblemTDEM_b(ProblemBaseTDEM):
     """
diff --git a/simpegEM/TDEM/__init__.py b/simpegEM/TDEM/__init__.py
index d489d934..8a64e946 100644
--- a/simpegEM/TDEM/__init__.py
+++ b/simpegEM/TDEM/__init__.py
@@ -1,3 +1,3 @@
 from BaseTDEM import ProblemBaseTDEM
-from SurveyTDEM import SurveyTDEM1D
+from SurveyTDEM import SurveyTDEM1D, FieldsTDEM
 from TDEM_b import ProblemTDEM_b