diff --git a/src/nomad_simulations/utils/utils.py b/src/nomad_simulations/utils/utils.py
index 0b8a2f3b..20178de8 100644
--- a/src/nomad_simulations/utils/utils.py
+++ b/src/nomad_simulations/utils/utils.py
@@ -130,7 +130,7 @@ def is_not_representative(model_system, logger: BoundLogger = None):
 
 def check_archive(archive: ArchiveSection, logger: BoundLogger = None):
     """
-    Checks if the given `Archive` is empty and logs a warning.
+    Checks if the given `ArchiveSection` is empty and logs a warning.
 
     Args:
         archive (ArchiveSection): The `ArchiveSection` to check.
diff --git a/tests/test_model_system.py b/tests/test_model_system.py
index 04a007a9..9e1a050d 100644
--- a/tests/test_model_system.py
+++ b/tests/test_model_system.py
@@ -24,7 +24,6 @@
 from . import logger
 from .conftest import get_template_atomic_cell
 from nomad_simulations.model_system import (
-    AtomicCell,
     Symmetry,
     ChemicalFormula,
     ModelSystem,