diff --git a/moor/test/ffi/vm_database_test.dart b/moor/test/ffi/vm_database_test.dart
index 633d96fb..4a041af5 100644
--- a/moor/test/ffi/vm_database_test.dart
+++ b/moor/test/ffi/vm_database_test.dart
@@ -1,5 +1,5 @@
+import 'package:moor/ffi.dart';
 import 'package:moor/moor.dart';
-import 'package:moor/src/ffi/vm_database.dart';
 import 'package:sqlite3/sqlite3.dart';
 import 'package:test/test.dart';
 
diff --git a/moor/test/integration_tests/regress_1235_test.dart b/moor/test/integration_tests/regress_1235_test.dart
index 1ba27a97..7a383f62 100644
--- a/moor/test/integration_tests/regress_1235_test.dart
+++ b/moor/test/integration_tests/regress_1235_test.dart
@@ -1,6 +1,6 @@
 import 'dart:async';
 
-import 'package:moor/src/ffi/vm_database.dart';
+import 'package:moor/ffi.dart';
 import 'package:test/scaffolding.dart';
 
 import '../data/tables/todos.dart';
diff --git a/moor/test/serialization_test.dart b/moor/test/serialization_test.dart
index f139e3de..9e8743a7 100644
--- a/moor/test/serialization_test.dart
+++ b/moor/test/serialization_test.dart
@@ -1,5 +1,4 @@
 import 'package:moor/moor.dart';
-import 'package:moor/src/runtime/data_class.dart';
 import 'package:test/test.dart';
 import 'data/tables/todos.dart';