diff --git a/tests/test_mymath.py b/tests/test_mymath.py
index bcc8e32..0d2d038 100644
--- a/tests/test_mymath.py
+++ b/tests/test_mymath.py
@@ -1,6 +1,6 @@
 import pytest
 import math as m
-from mymath import sinc2d
+from code.mymath import sinc2d
 
 @pytest.mark.parametrize("x,y,expected",
                          [(0,0,1),
@@ -10,4 +10,4 @@
                          ])
 def test_sinc2d(x,y,expected):
     observed = sinc2d(x,y)
-    assert expected == observed
\ No newline at end of file
+    assert expected == observed
diff --git a/tests/test_mymath2.py b/tests/test_mymath2.py
index d8e7262..c5c3b88 100644
--- a/tests/test_mymath2.py
+++ b/tests/test_mymath2.py
@@ -1,5 +1,5 @@
 import pytest
-from mymath import *
+from code.mymath import *
 
 @pytest.mark.parametrize("x,x_plus_one",[(0,1),(1,2),(1.5,2.5),(2.25,3.25)])
 def test_addone(x,x_plus_one):
@@ -15,4 +15,4 @@ def test_addone_then_double(x,x_plusone_doubled):
 
 @pytest.mark.parametrize("x,y,val",[(1,1,0.7080734182735712),(5,0.25,-0.18979332974154567)])
 def test_sinc2d_regression(x,y,val):
-    assert abs(sinc2d(x,y)-val) < 1e-10
\ No newline at end of file
+    assert abs(sinc2d(x,y)-val) < 1e-10