diff --git a/tests/test_op25_controller.py b/tests/test_op25_controller.py
index a76f46a..48228b1 100644
--- a/tests/test_op25_controller.py
+++ b/tests/test_op25_controller.py
@@ -82,6 +82,9 @@ class MockMetadataConfig(BaseModel):
 class MockTerminalConfig(BaseModel):
     pass
 
+class MockTalkgroupTag(BaseModel):
+    pass
+
 mock_models.ConfigGenerator = MockConfigGenerator
 mock_models.DecodeMode = MockDecodeMode
 mock_models.ChannelConfig = MockChannelConfig
@@ -91,6 +94,8 @@ mock_models.TrunkingChannelConfig = MockTrunkingChannelConfig
 mock_models.TerminalConfig = MockTerminalConfig
 mock_models.MetadataConfig = MockMetadataConfig
 mock_models.MetadataStreamConfig = MockMetadataStreamConfig
+mock_models.IcecastConfig = MockIcecastConfig
+mock_models.TalkgroupTag = MockTalkgroupTag
 
 sys.modules["models.models"] = mock_models
 sys.modules["models"] = types.ModuleType("models")