diff --git a/packages/ubuntu_provision_test/lib/src/init_tester.dart b/packages/ubuntu_provision_test/lib/src/init_tester.dart
index 98f22bf22..3d862d7f0 100644
--- a/packages/ubuntu_provision_test/lib/src/init_tester.dart
+++ b/packages/ubuntu_provision_test/lib/src/init_tester.dart
@@ -1,6 +1,7 @@
 import 'package:flutter_test/flutter_test.dart';
 import 'package:ubuntu_init/ubuntu_init.dart';
 import 'package:ubuntu_provision_test/src/wizard_tester.dart';
+import 'package:ubuntu_test/ubuntu_test.dart';
 import 'package:yaru_test/yaru_test.dart';
 
 extension UbuntuInitPageTester on WidgetTester {