diff --git a/tests/test_processing_info_maps.py b/tests/test_processing_info_maps.py
index 0ad939b..3dc5f86 100644
--- a/tests/test_processing_info_maps.py
+++ b/tests/test_processing_info_maps.py
@@ -6,7 +6,7 @@
 from nemosis import defaults
 import pandas as pd
 from datetime import datetime, timedelta
-from NEMOSIS.nemosis import query_wrappers
+from nemosis import query_wrappers
 
 
 class TestSearchTypeValidity(unittest.TestCase):
diff --git a/tests/test_query_wrappers.py b/tests/test_query_wrappers.py
index 51293e3..f1bb5e2 100644
--- a/tests/test_query_wrappers.py
+++ b/tests/test_query_wrappers.py
@@ -1,6 +1,6 @@
 import unittest
 import pandas as pd
-from NEMOSIS.nemosis import query_wrappers
+from nemosis import query_wrappers
 from pandas.testing import assert_frame_equal
 from datetime import datetime
 from nemosis import defaults