diff --git a/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py b/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py
index 50e3cda7..0e35eacf 100644
--- a/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py
+++ b/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py
@@ -50,7 +50,7 @@ midy = int(mesh.nCy/2)
 PF.Magnetics.plot_obs_2D(rxLoc,d,'Observed Data')
 
 #%% Run inversion
-prob = PF.Magnetics.MagneticIntegral(mesh, mapping=idenMap, actInd=actv)
+prob = PF.Magnetics.Problem3D_Integral(mesh, mapping=idenMap, actInd=actv)
 prob.solverOpts['accuracyTol'] = 1e-4
 survey.pair(prob)
 #survey.makeSyntheticData(data, std=0.01)
diff --git a/simpegPF/Magnetics.py b/simpegPF/Magnetics.py
index 27ca979a..64fa6363 100644
--- a/simpegPF/Magnetics.py
+++ b/simpegPF/Magnetics.py
@@ -4,7 +4,7 @@ from scipy.constants import mu_0
 from MagAnalytics import spheremodel, CongruousMagBC
 import re
 
-class MagneticIntegral(Problem.BaseProblem):
+class Problem3D_Integral(Problem.BaseProblem):
 
     #surveyPair = Survey.LinearSurvey
 
@@ -206,7 +206,7 @@ class MagneticIntegral(Problem.BaseProblem):
 
         return G
 
-class MagneticsDiffSecondary(Problem.BaseProblem):
+class Problem3D_DiffSecondary(Problem.BaseProblem):
     """
         Secondary field approach using differential equations!
     """