diff --git a/src/main/java/com/primefactorsolutions/views/WorkDocumentsView.java b/src/main/java/com/primefactorsolutions/views/WorkDocumentsView.java
index dfad0df..d0ec2b0 100644
--- a/src/main/java/com/primefactorsolutions/views/WorkDocumentsView.java
+++ b/src/main/java/com/primefactorsolutions/views/WorkDocumentsView.java
@@ -81,7 +81,7 @@ public class WorkDocumentsView extends Main {
     }
     private Button createSaveButton(String sectionTitle, Button viewButton) {
         Button saveButton = new Button("Save");
-        saveButton.addClickListener(clickEvent -> saveFile(sectionTitle, ));
+        saveButton.addClickListener(clickEvent -> saveFile(sectionTitle, viewButton));
         return saveButton;
     }
     private HorizontalLayout createLayoutWithComponent(com.vaadin.flow.component.Component component) {