diff --git a/src/QueryInterceptor.php b/src/QueryInterceptor.php
index 3379656..2c32d19 100644
--- a/src/QueryInterceptor.php
+++ b/src/QueryInterceptor.php
@@ -64,8 +64,9 @@ private function getQueryResult(MethodInvocation $invocation, QueryInterface $qu
     }
 
     /**
-     * @param MethodInvocation<object> $invocation
-     * @param mixed $result
+     * @param MethodInvocation $invocation
+     * @param array|object|scalar|null $result
+     * @return ResourceObject
      */
     private function returnRo(ResourceObject $ro, MethodInvocation $invocation, $result): ResourceObject
     {