diff --git a/tests/em/tdem/test_TDEM_forward_Analytic.py b/tests/em/tdem/test_TDEM_forward_Analytic.py
index 750f3c2e..1d5b30ec 100644
--- a/tests/em/tdem/test_TDEM_forward_Analytic.py
+++ b/tests/em/tdem/test_TDEM_forward_Analytic.py
@@ -63,10 +63,10 @@ def halfSpaceProblemAnaDiff(meshType, sig_half=1e-2, rxOffset=50., bounds=None,
     return log10diff
 
 
-# class TDEM_SimpleSrcTests(unittest.TestCase):
-#     def test_source(self):
-#         waveform = EM.TDEM.SurveyTDEM.StepOffWaveform()
-#         assert waveform.eval(0.) == 0.
+class TDEM_SimpleSrcTests(unittest.TestCase):
+    def test_source(self):
+        waveform = EM.TDEM.Src.StepOffWaveform()
+        assert waveform.eval(0.) == 0.
 
 
 class TDEM_bTests(unittest.TestCase):