diff --git a/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php b/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php
index b68b010..8dc1178 100644
--- a/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php
+++ b/src/Type/PHPUnit/CreateMockDynamicReturnTypeExtension.php
@@ -22,6 +22,7 @@ class CreateMockDynamicReturnTypeExtension implements \PHPStan\Type\DynamicMetho
 		'createTestProxy' => 0,
 		'getMockForAbstractClass' => 0,
 		'getMockFromWsdl' => 1,
+		'getMockForTrait' => 0,
 	];
 
 	public function getClass(): string