diff --git a/test/cubic2d.py b/test/cubic2d.py
index 64c4694..286ab28 100644
--- a/test/cubic2d.py
+++ b/test/cubic2d.py
@@ -4,7 +4,7 @@ import unittest
 
 from interp.baker        import run_baker
 
-from interp.grid.DD      import grid
+from interp.grid         import grid
 from interp.grid.simplex import contains
 
 def exact_func(X):
diff --git a/test/quadratic2d.py b/test/quadratic2d.py
index bec768d..1882799 100644
--- a/test/quadratic2d.py
+++ b/test/quadratic2d.py
@@ -4,7 +4,7 @@ import unittest
 
 from interp.baker        import run_baker
 
-from interp.grid.DD      import grid
+from interp.grid         import grid
 from interp.grid.simplex import contains
 
 def exact_func(X):