diff --git a/framework-client/src/main/java/io/axoniq/platform/framework/modelling/AxoniqPlatformStateManager.kt b/framework-client/src/main/java/io/axoniq/platform/framework/modelling/AxoniqPlatformStateManager.kt
index 1041add..8fcb431 100644
--- a/framework-client/src/main/java/io/axoniq/platform/framework/modelling/AxoniqPlatformStateManager.kt
+++ b/framework-client/src/main/java/io/axoniq/platform/framework/modelling/AxoniqPlatformStateManager.kt
@@ -49,7 +49,7 @@ class AxoniqPlatformStateManager(
         return delegate.registeredIdsFor(entityType)
     }
 
-    override fun <ID : Any, T : Any> repository(entityType: Class<T>, idType: Class<ID>): Repository<ID, T> {
+    override fun <ID : Any, T : Any> repository(entityType: Class<T>, idType: Class<ID>): Repository<ID, T>? {
         return delegate.repository(entityType, idType)
     }
 
diff --git a/pom.xml b/pom.xml
index 37d76cc..9314fe2 100644
--- a/pom.xml
+++ b/pom.xml
@@ -57,7 +57,7 @@
         <kotlin.version>2.2.20</kotlin.version>
         <dokka.version>2.1.0</dokka.version>
 
-        <axoniq-framework.version>5.3.0</axoniq-framework.version>
+        <axoniq-framework.version>5.3.2</axoniq-framework.version>
         <jackson.version>3.1.0</jackson.version>
         <jackson-annotations.version>2.21</jackson-annotations.version>
         <spring-boot.version>3.5.7</spring-boot.version>