diff --git a/tests/test_atoms_state.py b/tests/test_atoms_state.py
index 9f30e2bb..27d80903 100644
--- a/tests/test_atoms_state.py
+++ b/tests/test_atoms_state.py
@@ -22,7 +22,7 @@
 
 from nomad.units import ureg
 
-from simulationdataschema.atoms_state import (
+from nomad_simulations.atoms_state import (
     OrbitalsState,
     CoreHole,
     HubbardInteractions,
diff --git a/tests/test_model_system.py b/tests/test_model_system.py
index 2fae4e20..20299a19 100644
--- a/tests/test_model_system.py
+++ b/tests/test_model_system.py
@@ -18,7 +18,7 @@
 
 import pytest
 
-from simulationdataschema.model_system import AtomicCell
+from nomad_simulations.model_system import AtomicCell
 
 
 class TestAtomicCell: