1 files changed, 3 insertions, 2 deletions
diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodesV2ApiHandler.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodesV2ApiHandler.java
index 0875bf3815d..c850962bf53 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodesV2ApiHandler.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodesV2ApiHandler.java
@@ -392,9 +392,10 @@ public class NodesV2ApiHandler extends LoggingRequestHandler {
                                                                    throw new IllegalArgumentException("Invalid duration '" + s + "'", e);
                                                                }
                                                            });
-                nodeRepository.osVersions().setTarget(nodeType, osVersion, upgradeBudget, force);
+                if (upgradeBudget.isEmpty()) throw new IllegalArgumentException("upgradeBudget must be set");
+                nodeRepository.osVersions().setTarget(nodeType, osVersion, upgradeBudget.get(), force);
                 messageParts.add("osVersion to " + osVersion.toFullString());
-                upgradeBudget.ifPresent(d -> messageParts.add("upgradeBudget to " + d));
+                messageParts.add("upgradeBudget to " + upgradeBudget.get());
             }
         }