diff --git a/tests/integration_tests/extensions/services/extended_service.py b/tests/integration_tests/extensions/services/extended_service.py
index ddaf4a1e..ac58091c 100644
--- a/tests/integration_tests/extensions/services/extended_service.py
+++ b/tests/integration_tests/extensions/services/extended_service.py
@@ -17,13 +17,15 @@ from primaite.utils.validation.port import PORT_LOOKUP
 _LOGGER = getLogger(__name__)
 
 
-class ExtendedService(Service, identifier="extendedservice"):
+class ExtendedService(Service, identifier="ExtendedService"):
     """
     A copy of DatabaseService that uses the extension framework instead of being part of PrimAITE.
 
     This class inherits from the `Service` class and provides methods to simulate a SQL database.
     """
 
+    config: "ExtendedService.ConfigSchema" = None
+
     password: Optional[str] = None
     """Password that needs to be provided by clients if they want to connect to the DatabaseService."""
 
@@ -36,6 +38,11 @@ class ExtendedService(Service, identifier="extendedservice"):
     latest_backup_file_name: str = None
     """File name of latest backup."""
 
+    class ConfigSchema(Service.ConfigSchema):
+        """ConfigSchema for ExtendedService."""
+
+        type: str = "EXTENDED_SERVICE"
+
     def __init__(self, **kwargs):
         kwargs["name"] = "ExtendedService"
         kwargs["port"] = PORT_LOOKUP["POSTGRES_SERVER"]
diff --git a/tests/unit_tests/_primaite/_simulator/_system/test_software.py b/tests/unit_tests/_primaite/_simulator/_system/test_software.py
index 300f8d9d..d9da7d73 100644
--- a/tests/unit_tests/_primaite/_simulator/_system/test_software.py
+++ b/tests/unit_tests/_primaite/_simulator/_system/test_software.py
@@ -10,7 +10,15 @@ from primaite.utils.validation.ip_protocol import PROTOCOL_LOOKUP
 from primaite.utils.validation.port import PORT_LOOKUP
 
 
-class TestSoftware(Service):
+class TestSoftware(Service, identifier="TestSoftware"):
+
+    config: "TestSoftware.ConfigSchema" = None
+
+    class ConfigSchema(Service.ConfigSchema):
+        """ConfigSChema for TestSoftware."""
+
+        type: str = "TEST_SOFTWARE"
+
     def describe_state(self) -> Dict:
         pass