diff --git a/lib/data/models/design_model.dart b/lib/data/models/design_model.dart
index 8adca75ea..cf036d9ad 100644
--- a/lib/data/models/design_model.dart
+++ b/lib/data/models/design_model.dart
@@ -67,6 +67,7 @@ abstract class DesignEntity extends Object
 
     if (response == 0) {
       // STARTER: sort default - do not remove comment
+      return 0;
     } else {
       return response;
     }
diff --git a/lib/ui/design/design_list_vm.dart b/lib/ui/design/design_list_vm.dart
index c06dd8f96..ae431600a 100644
--- a/lib/ui/design/design_list_vm.dart
+++ b/lib/ui/design/design_list_vm.dart
@@ -1,5 +1,6 @@
 import 'dart:async';
 import 'package:invoiceninja_flutter/data/models/design_model.dart';
+import 'package:invoiceninja_flutter/ui/design/design_presenter.dart';
 import 'package:redux/redux.dart';
 import 'package:invoiceninja_flutter/redux/app/app_actions.dart';
 import 'package:flutter/material.dart';
@@ -7,7 +8,6 @@ import 'package:flutter/foundation.dart';
 import 'package:flutter/widgets.dart';
 import 'package:flutter_redux/flutter_redux.dart';
 import 'package:built_collection/built_collection.dart';
-import 'package:invoiceninja_flutter/redux/client/client_actions.dart';
 import 'package:invoiceninja_flutter/redux/ui/list_ui_state.dart';
 import 'package:invoiceninja_flutter/utils/completers.dart';
 import 'package:invoiceninja_flutter/utils/localization.dart';
@@ -90,6 +90,7 @@ class DesignListVM {
               EntityAction action) =>
           handleDesignAction(context, designs, action),
       onRefreshed: (context) => _handleRefresh(context),
+      tableColumns: DesignPresenter.getTableFields(state.userCompany),
     );
   }
 
diff --git a/stubs/ui/stub/stub_list_vm b/stubs/ui/stub/stub_list_vm
index 2a7d3e745..0a3422dbf 100644
--- a/stubs/ui/stub/stub_list_vm
+++ b/stubs/ui/stub/stub_list_vm
@@ -1,13 +1,13 @@
 import 'dart:async';
 import 'package:invoiceninja_flutter/data/models/stub_model.dart';
 import 'package:redux/redux.dart';
+import 'package:invoiceninja_flutter/ui/design/design_presenter.dart';
 import 'package:invoiceninja_flutter/redux/app/app_actions.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter/foundation.dart';
 import 'package:flutter/widgets.dart';
 import 'package:flutter_redux/flutter_redux.dart';
 import 'package:built_collection/built_collection.dart';
-import 'package:invoiceninja_flutter/redux/client/client_actions.dart';
 import 'package:invoiceninja_flutter/redux/ui/list_ui_state.dart';
 import 'package:invoiceninja_flutter/utils/completers.dart';
 import 'package:invoiceninja_flutter/utils/localization.dart';
@@ -92,6 +92,7 @@ class StubListVM {
           (BuildContext context, List<BaseEntity> stubs, EntityAction action) =>
           handleStubAction(context, stubs, action),
         onRefreshed: (context) => _handleRefresh(context),
+        tableColumns: StubPresenter.getTableFields(state.userCompany),
     );
   }