1 files changed, 4 insertions, 3 deletions
diff --git a/src/test/test_hydrilla_builder.py b/src/test/test_hydrilla_builder.py
index f4a4d2f..410b7a1 100644
--- a/src/test/test_hydrilla_builder.py
+++ b/src/test/test_hydrilla_builder.py
@@ -16,7 +16,8 @@ from typing import Callable, Optional
 
 from jsonschema import ValidationError
 
-from hydrilla_builder import build
+from hydrilla import util as hydrilla_util
+from hydrilla.builder import build
 
 here = Path(__file__).resolve().parent
 
@@ -187,7 +188,7 @@ def prepare_modified(tmpdir: Path, modify_cb: ModifyCb) -> CaseSettings:
     settings.srcdir = tmpdir / 'srcdir_copy'
 
     with open(settings.srcdir / 'index.json', 'rt') as file_handle:
-        obj = json.loads(build.strip_json_comments(file_handle.read()))
+        obj = json.loads(hydrilla_util.strip_json_comments(file_handle.read()))
 
     contents = modify_cb(settings, obj)
 
@@ -438,4 +439,4 @@ def test_build_error(tmpdir: str, break_index_json: tuple[ModifyCb, type]):
 
     with pytest.raises(error_type):
         build.Build(settings.srcdir, settings.index_json_path)\
-             .write_package_files(dstdir)
+            .write_package_files(dstdir)