diff --git a/code/test_array_stats_pytest.py b/code/test_array_stats_pytest.py
index f3047e1..1616843 100644
--- a/code/test_array_stats_pytest.py
+++ b/code/test_array_stats_pytest.py
@@ -1,5 +1,5 @@
 import pytest
-import code.array_stats
+import array_stats
 
 @pytest.mark.parametrize("data_list,expected",
                          [([-2, -1, 0, 1, 2],0),
diff --git a/code/test_mymath.py b/code/test_mymath.py
index 0d2d038..7fd44a3 100644
--- a/code/test_mymath.py
+++ b/code/test_mymath.py
@@ -1,6 +1,6 @@
 import pytest
 import math as m
-from code.mymath import sinc2d
+from mymath import sinc2d
 
 @pytest.mark.parametrize("x,y,expected",
                          [(0,0,1),
diff --git a/code/test_mymath2.py b/code/test_mymath2.py
index c5c3b88..24f8282 100644
--- a/code/test_mymath2.py
+++ b/code/test_mymath2.py
@@ -1,5 +1,5 @@
 import pytest
-from code.mymath import *
+from 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):