diff --git a/lib/redux/app/app_state.dart b/lib/redux/app/app_state.dart
index c399e70d5..7553ee0c9 100644
--- a/lib/redux/app/app_state.dart
+++ b/lib/redux/app/app_state.dart
@@ -11,6 +11,7 @@ import 'package:invoiceninja_flutter/redux/company/company_state.dart';
 import 'package:invoiceninja_flutter/redux/company_gateway/company_gateway_selectors.dart';
 import 'package:invoiceninja_flutter/redux/credit/credit_selectors.dart';
 import 'package:invoiceninja_flutter/redux/dashboard/dashboard_state.dart';
+import 'package:invoiceninja_flutter/redux/design/design_state.dart';
 import 'package:invoiceninja_flutter/redux/document/document_state.dart';
 import 'package:invoiceninja_flutter/redux/expense/expense_selectors.dart';
 import 'package:invoiceninja_flutter/redux/expense/expense_state.dart';
diff --git a/starter.sh b/starter.sh
index 488b1f34d..2f44e6311 100644
--- a/starter.sh
+++ b/starter.sh
@@ -185,7 +185,7 @@ else
 
     # Link in new module
     comment="STARTER: import - do not remove comment"
-    code="import 'package:${package}\/redux\/${module_snake}\/${module_snake}_state.dart';import 'package:invoiceninja_flutter/ui/${module_snake}/edit/${module_snake}_edit_vm.dart';${lineBreak}"
+    code="import 'package:${package}\/redux\/${module_snake}\/${module_snake}_state.dart';import 'package:invoiceninja_flutter/ui/${module_snake}/edit/${module_snake}_edit_vm.dart';import 'package:invoiceninja_flutter/redux/${module_snake}/${module_snake}_state.dart';${lineBreak}"
     sed -i -e "s/$comment/$comment${lineBreak}$code/g" ./lib/redux/app/app_state.dart
 
     comment="STARTER: states switch list - do not remove comment"