diff --git a/lib/main.dart b/lib/main.dart
index 332cc3dee..ddbfd1110 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -36,6 +36,7 @@ import 'package:redux/redux.dart';
 import 'package:redux_logging/redux_logging.dart';
 import 'package:sentry/sentry.dart';
 import 'package:shared_preferences/shared_preferences.dart';
+// STARTER: import - do not remove comment
 
 void main({bool isTesting = false}) async {
   WidgetsFlutterBinding.ensureInitialized();
diff --git a/lib/main_app.dart b/lib/main_app.dart
index 6166398e9..a6453f245 100644
--- a/lib/main_app.dart
+++ b/lib/main_app.dart
@@ -43,7 +43,7 @@ import 'package:invoiceninja_flutter/utils/localization.dart';
 import 'package:invoiceninja_flutter/utils/platforms.dart';
 import 'package:local_auth/local_auth.dart';
 import 'package:redux/redux.dart';
-
+// STARTER: import - do not remove comment
 
 class InvoiceNinjaApp extends StatefulWidget {
   const InvoiceNinjaApp({Key key, this.store}) : super(key: key);
diff --git a/starter.sh b/starter.sh
index f7e75652a..46d05c450 100644
--- a/starter.sh
+++ b/starter.sh
@@ -314,11 +314,18 @@ else
     code="..addAll(createStore${Module}sMiddleware())${lineBreak}"
     sed -i -e "s/$comment/$comment${lineBreak}$code/g" ./lib/main.dart
 
+    comment="STARTER: import - do not remove comment"
+    code="import 'package:${package}\/ui\/${module_snake}\/${module_snake}_screen.dart';${lineBreak}"
+    code="${code}import 'package:${package}\/ui\/${module_snake}\/edit\/${module_snake}_edit_vm.dart';${lineBreak}"
+    code="${code}import 'package:${package}\/ui\/${module_snake}\/view\/${module_snake}_view_vm.dart';${lineBreak}"
+    code="${code}import 'package:${package}\/ui\/${module_snake}\/${module_snake}_screen_vm.dart';${lineBreak}"
+    sed -i -e "s/$comment/$comment${lineBreak}$code/g" ./lib/main.dart
+
     comment="STARTER: routes - do not remove comment"
     code="${Module}Screen.route: (context) => ${Module}ScreenBuilder(),${lineBreak}"
     code="${code}${Module}ViewScreen.route: (context) => ${Module}ViewScreen(),${lineBreak}"
     code="${code}${Module}EditScreen.route: (context) => ${Module}EditScreen(),${lineBreak}"
-    sed -i -e "s/$comment/$comment${lineBreak}$code/g" ./lib/main.dart
+    sed -i -e "s/$comment/$comment${lineBreak}$code/g" ./lib/main_app.dart
 
     comment="STARTER: import - do not remove comment"
     code="import 'package:${package}\/data\/models\/${module_snake}_model.dart';${lineBreak}"