diff --git a/src/test/python/release_mixin/mock_infrastructure.py b/src/test/python/release_mixin/mock_infrastructure.py
index 74bd1a5..9cf6b06 100644
--- a/src/test/python/release_mixin/mock_infrastructure.py
+++ b/src/test/python/release_mixin/mock_infrastructure.py
@@ -1,6 +1,6 @@
 from pathlib import Path
 
-from src.main.python.ddadevops.domain import ReleaseType, Version
+from src.main.python.ddadevops.domain import ReleaseType, Version, Release
 
 from .mock_infrastructure_api import MockGitApi
 
@@ -38,6 +38,6 @@ class MockReleaseRepository():
         self.main_branch = main_branch
         self.get_release_count = 0
 
-    def get_release(self) -> :
+    def get_release(self) -> Release:
         self.get_release_count += 1
         return Release(self.release_type_repository.get_release_type(), self.version_repository.get_version(), self.main_branch)