diff --git a/simpegEM/FDEM/FieldsFDEM.py b/simpegEM/FDEM/FieldsFDEM.py
index 1917a379..2e3e10d8 100644
--- a/simpegEM/FDEM/FieldsFDEM.py
+++ b/simpegEM/FDEM/FieldsFDEM.py
@@ -161,8 +161,6 @@ class FieldsFDEM_h(FieldsFDEM):
         self._edgeCurl = self.survey.prob.mesh.edgeCurl
         self._MeMuI = self.survey.prob.MeMuI
         self._MfSigmai = self.survey.prob.MfSigmai
-        # self._getSource = self.survey.prob.getSource
-        # self._getSourceDeriv = self.survey.prob.getSourceDeriv 
 
     def _j_sec(self, h, src): # adjoint=False
         return self._edgeCurl*h
diff --git a/simpegEM/FDEM/SurveyFDEM.py b/simpegEM/FDEM/SurveyFDEM.py
index 595e0403..90a04ac5 100644
--- a/simpegEM/FDEM/SurveyFDEM.py
+++ b/simpegEM/FDEM/SurveyFDEM.py
@@ -377,7 +377,7 @@ class SurveyFDEM(Survey.BaseSurvey):
         if getattr(self, '_nSrcByFreq', None) is None:
             self._nSrcByFreq = {}
             for freq in self.freqs:
-                self._nSrcByFreq[freq] = len(self.getSource(freq))
+                self._nSrcByFreq[freq] = len(self.getSrcByFreq(freq))
         return self._nSrcByFreq
 
     def getSrcByFreq(self, freq):