diff --git a/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php b/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php
index 68038e0..6d74e3f 100644
--- a/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php
+++ b/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php
@@ -16,6 +16,8 @@ class CreateMockDynamicReturnTypeExtension implements \PHPStan\Type\DynamicMetho
 	private $methods = [
 		'createMock' => 0,
 		'createConfiguredMock' => 0,
+		'createPartialMock' => 0,
+		'createTestProxy' => 0,
 		'getMockForAbstractClass' => 0,
 		'getMockFromWsdl' => 1,
 	];