diff --git a/tests/test_messages.py b/tests/test_messages.py
index 198fbb9c0d9d46094174682c68fbc396f72ceb1f..bdbf371f6961e1f46fe1577ea2134f918e4588de 100644
--- a/tests/test_messages.py
+++ b/tests/test_messages.py
@@ -29,8 +29,7 @@
 
 def test_messages_dict_behavior():
 
-    from caosdb.common.models import Message
-    from caosdb.common.models import _Messages
+    from caosdb.common.models import Message, Messages
 
     msgs = _Messages()