diff --git a/tests/unit/v0/test_data_models.py b/tests/unit/v0/test_data_models.py
index e5621b8a..7ea7ce68 100644
--- a/tests/unit/v0/test_data_models.py
+++ b/tests/unit/v0/test_data_models.py
@@ -173,7 +173,7 @@ def test_action_params_parsing_ok(self):
         mock_event.params = {"host": "my-host"}
         with self.assertLogs(level="INFO") as logger:
             self.assertTrue(self.harness.charm._set_server_action(mock_event))
-        self.assertEqual(sorted(logger.output), ["INFO:unit.test_data_models:my-host:80"])
+        self.assertEqual(sorted(logger.output), ["INFO:test_data_models:my-host:80"])
 
     def test_action_params_parsing_ko(self):
         """Test that action parameters validation would raise an exception."""
@@ -198,7 +198,7 @@ def test_relation_databag_io(self):
 
         with self.assertLogs(level="INFO") as logger:
             self.harness.update_relation_data(relation_id, "mongodb", {"key": "1.0"})
-        self.assertEqual(logger.output, ["INFO:unit.test_data_models:Field type: <class 'float'>"])
+        self.assertEqual(logger.output, ["INFO:test_data_models:Field type: <class 'float'>"])
 
     def test_relation_databag_merged(self):
         """Test that relation databag of unit and app can be read and merged into a single pydantic object."""