diff --git a/lib/ui/project/view/project_view_vm.dart b/lib/ui/project/view/project_view_vm.dart
index 2b6e05ed0..85009430f 100644
--- a/lib/ui/project/view/project_view_vm.dart
+++ b/lib/ui/project/view/project_view_vm.dart
@@ -53,7 +53,10 @@ class ProjectViewVM {
     final project = state.projectState.map[state.projectUIState.selectedId];
 
     return ProjectViewVM(
-        isLoading: store.state.isLoading,
+        company: state.selectedCompany,
+        isSaving: state.isSaving,
+        isLoading: state.isLoading,
+        isDirty: project.isNew,
         project: project,
         onEditPressed: (BuildContext context) {
           store.dispatch(EditProject(project: project, context: context));
diff --git a/stubs/ui/stub/view/stub_view_vm b/stubs/ui/stub/view/stub_view_vm
index 6121ba37a..842f7ba8e 100644
--- a/stubs/ui/stub/view/stub_view_vm
+++ b/stubs/ui/stub/view/stub_view_vm
@@ -53,7 +53,10 @@ class StubViewVM {
     final stub = state.stubState.map[state.stubUIState.selectedId];
 
     return StubViewVM(
-        isLoading: store.state.isLoading,
+        company: state.selectedCompany,
+        isSaving: state.isSaving,
+        isLoading: state.isLoading,
+        isDirty: stub.isNew,
         stub: stub,
         onEditPressed: (BuildContext context) {
           store.dispatch(EditStub(stub: stub, context: context));