1 files changed, 9 insertions, 3 deletions
diff --git a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/userInterface/CreateOperationController.java b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/userInterface/CreateOperationController.java
index 9debddf..67c8807 100644
--- a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/userInterface/CreateOperationController.java
+++ b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/userInterface/CreateOperationController.java
@@ -182,7 +182,7 @@ public class CreateOperationController {
         }
     }
 
-    private static ContextMenu createContextMenu(Vehicle data, VehicleService vehicleService) {
+    private ContextMenu createContextMenu(Vehicle data, VehicleService vehicleService) {
         ContextMenu menu = new ContextMenu();
 
         for (Vehicle.Status status : Vehicle.Status.values()) {
@@ -202,6 +202,7 @@ public class CreateOperationController {
                     event -> {
                         try {
                             vehicleService.update(data.toBuilder().status(status).build());
+                            this.updateList();
                         } catch (InvalidVehicleException | ServiceException e) {
                             LOG.error("Error while setting status.", e);
                             Alert a = new Alert(AlertType.ERROR, e.getMessage());
@@ -236,7 +237,12 @@ public class CreateOperationController {
                         }
 
                         vehicleService.update(
-                                data.toBuilder().registrations(newRegistrations).build());
+                                data.toBuilder()
+                                        .registrations(newRegistrations)
+                                        .status(Vehicle.Status.ABGEMELDET)
+                                        .build());
+
+                        this.updateList();
                     } catch (InvalidVehicleException | ServiceException e) {
                         LOG.error("Error while unregistering.", e);
                         Alert a = new Alert(AlertType.ERROR, e.getMessage());
@@ -342,7 +348,7 @@ public class CreateOperationController {
         updateList();
     }
 
-    public void setVisible(boolean b) {
+    void setVisible(boolean b) {
         apInvisible.setVisible(!b);
     }