diff --git a/composer.json b/composer.json
index b0cdc1f..b84a811 100644
--- a/composer.json
+++ b/composer.json
@@ -19,7 +19,7 @@
 		"jakub-onderka/php-parallel-lint": "^1.0",
 		"phing/phing": "^2.16.0",
 		"phpstan/phpstan-strict-rules": "^0.12",
-		"phpunit/phpunit": "^7.0",
+		"phpunit/phpunit": "^7.0 || ^8.0 || ^9.0",
 		"satooshi/php-coveralls": "^1.0",
 		"slevomat/coding-standard": "^4.7.2"
 	},
diff --git a/src/Rules/PHPUnit/MockMethodCallRule.php b/src/Rules/PHPUnit/MockMethodCallRule.php
index 8a00af6..d8bcd8a 100644
--- a/src/Rules/PHPUnit/MockMethodCallRule.php
+++ b/src/Rules/PHPUnit/MockMethodCallRule.php
@@ -8,7 +8,7 @@
 use PHPStan\Type\Generic\GenericObjectType;
 use PHPStan\Type\IntersectionType;
 use PHPStan\Type\ObjectType;
-use PHPUnit\Framework\MockObject\InvocationMocker;
+use PHPUnit\Framework\MockObject\Builder\InvocationMocker;
 use PHPUnit\Framework\MockObject\MockObject;
 
 /**
diff --git a/src/Type/PHPUnit/MockObjectDynamicReturnTypeExtension.php b/src/Type/PHPUnit/MockObjectDynamicReturnTypeExtension.php
index 828fac7..bd8b00a 100644
--- a/src/Type/PHPUnit/MockObjectDynamicReturnTypeExtension.php
+++ b/src/Type/PHPUnit/MockObjectDynamicReturnTypeExtension.php
@@ -9,7 +9,7 @@
 use PHPStan\Type\IntersectionType;
 use PHPStan\Type\Type;
 use PHPStan\Type\TypeWithClassName;
-use PHPUnit\Framework\MockObject\InvocationMocker;
+use PHPUnit\Framework\MockObject\Builder\InvocationMocker;
 use PHPUnit\Framework\MockObject\MockObject;
 
 class MockObjectDynamicReturnTypeExtension implements \PHPStan\Type\DynamicMethodReturnTypeExtension