diff --git a/generator/tests/MethodTest.php b/generator/tests/MethodTest.php
index 0cfe4d4e..9ef269ed 100644
--- a/generator/tests/MethodTest.php
+++ b/generator/tests/MethodTest.php
@@ -45,7 +45,7 @@ public function testGetTypeHintFromResource(): void
         $params = $method->getParams();
         $this->assertEquals('string', $params[0]->getDocBlockType());
         $this->assertEquals('string', $params[0]->getSignatureType());
-        $this->assertEquals('string|int|float|bool', $params[1]->getDocBlockType());
+        $this->assertEquals('string|int|float|bool|null', $params[1]->getDocBlockType());
         $this->assertTrue($params[1]->isVariadic());
         $this->assertEquals('', $params[1]->getSignatureType());
 
@@ -54,7 +54,7 @@ public function testGetTypeHintFromResource(): void
         $method = new Method($xmlObject[0], $docPage->loadAndResolveFile(), $docPage->getModule(), new PhpStanFunctionMapReader(), Method::FALSY_TYPE);
         $params = $method->getParams();
         $this->assertEquals('string', $params[0]->getDocBlockType());
-        $this->assertEquals('callable', $params[1]->getDocBlockType());
+        $this->assertEquals('callable(array<int|string, string>):string', $params[1]->getDocBlockType());
         $this->assertEquals('string', $params[0]->getSignatureType());
         $this->assertEquals('callable', $params[1]->getSignatureType());