diff --git a/code/utils/linear_modeling.py b/code/utils/linear_modeling.py
index c9116cf..7eaa077 100644
--- a/code/utils/linear_modeling.py
+++ b/code/utils/linear_modeling.py
@@ -5,7 +5,7 @@
 import numpy.linalg as npl
 from scipy.stats import t as t_dist
 from scipy.ndimage import gaussian_filter
-import scene_slicer as ssm
+from . import scene_slicer as ssm
 import nibabel as nib
 
 
diff --git a/code/utils/tests/test_linear_modeling.py b/code/utils/tests/test_linear_modeling.py
index e3d9d77..db691d4 100644
--- a/code/utils/tests/test_linear_modeling.py
+++ b/code/utils/tests/test_linear_modeling.py
@@ -68,7 +68,7 @@ def test_get_top_32():
 
 def test_get_index_4d():
     actual = linear_modeling.get_index_4d([0, 2], data)
-    assert_equal(actual, [(0, 0), (0, 2)])
+    assert_equal(list(actual), [(0, 0), (0, 2)])
 
     # def test_plot_single_voxel():
     # 	not sure how to test plots