diff --git a/SimPEG/Tests.py b/SimPEG/Tests.py
index 4fe7f1b9..95f37c06 100644
--- a/SimPEG/Tests.py
+++ b/SimPEG/Tests.py
@@ -4,7 +4,7 @@ from numpy.linalg import norm
 from SimPEG.Utils import mkvc, sdiag, diagEst
 from SimPEG import Utils
 from SimPEG.Mesh import TensorMesh, CurvilinearMesh, CylMesh
-from SimPEG.Mesh.TreeMesh import TreeMesh
+from SimPEG.Mesh.TreeMesh import TreeMesh as Tree
 import numpy as np
 import scipy.sparse as sp
 import unittest