diff --git a/src/ribasim_nl/tests/test_code_utils.py b/src/ribasim_nl/tests/test_code_utils.py
index 9f3371aa..61774653 100644
--- a/src/ribasim_nl/tests/test_code_utils.py
+++ b/src/ribasim_nl/tests/test_code_utils.py
@@ -1,5 +1,5 @@
 """Tests for hydamo.code_utils"""
-from ribasim.utils import codes
+from ribasim_nl import codes
 
 
 def test_init():