diff --git a/stubs/data/models/stub_model b/stubs/data/models/stub_model
index d8fa9ffa2..137880fc1 100644
--- a/stubs/data/models/stub_model
+++ b/stubs/data/models/stub_model
@@ -1,6 +1,7 @@
 import 'package:built_value/built_value.dart';
+import 'package:built_collection/built_collection.dart';
 import 'package:built_value/serializer.dart';
-import 'package:invoiceninja_flutter/data/models/models.dart';
+import 'package:invoiceninja_flutter/data/models/stub_model.dart';
 
 part 'stub_model.g.dart';