diff --git a/lib/src/mock.dart b/lib/src/mock.dart
index 7601072..43b2ec0 100644
--- a/lib/src/mock.dart
+++ b/lib/src/mock.dart
@@ -17,7 +17,7 @@
 import 'package:meta/meta.dart';
 import 'package:mockito/src/call_pair.dart';
 import 'package:mockito/src/invocation_matcher.dart';
-import 'package:test/test.dart';
+import 'package:test_api/test_api.dart';
 
 bool _whenInProgress = false;
 bool _untilCalledInProgress = false;
diff --git a/pubspec.yaml b/pubspec.yaml
index b7e2307..45ef09b 100644
--- a/pubspec.yaml
+++ b/pubspec.yaml
@@ -15,9 +15,10 @@
   collection: ^1.1.0
   matcher: ^0.12.3
   meta: ^1.0.4
-  test: ^1.2.0
+  test_api: ^0.1.1
 
 dev_dependencies:
+  test: ^1.4.0
   build_runner: ^1.0.0
   build_test: ^0.10.0
   build_web_compilers: ^0.4.0