diff --git a/pom.xml b/pom.xml
index d2e359186..010da0e1f 100644
--- a/pom.xml
+++ b/pom.xml
@@ -10,7 +10,7 @@
         <version.junit5>5.10.3</version.junit5>
         <version.okhttp>4.12.0</version.okhttp>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
-        <version.unleash.specification>5.1.7</version.unleash.specification>
+        <version.unleash.specification>5.1.9</version.unleash.specification>
         <arguments />
         <version.jackson>2.17.2</version.jackson>
         <version.logback>1.3.14</version.logback>
diff --git a/src/main/java/io/getunleash/DefaultUnleash.java b/src/main/java/io/getunleash/DefaultUnleash.java
index abf82cc3d..59e1aaab2 100644
--- a/src/main/java/io/getunleash/DefaultUnleash.java
+++ b/src/main/java/io/getunleash/DefaultUnleash.java
@@ -106,7 +106,7 @@ public DefaultUnleash(
                     try {
                         this.unleashEngine.takeState(
                                 JsonFeatureParser.toJsonString(featureCollection));
-                    } catch (YggdrasilInvalidInputException e) {
+                    } catch (YggdrasilInvalidInputException | YggdrasilError e) {
                         LOGGER.error("Unable to update features", e);
                     }
                 });
diff --git a/src/test/java/io/getunleash/repository/UnleashEngineStateHandler.java b/src/test/java/io/getunleash/repository/UnleashEngineStateHandler.java
index 5b7ad6b95..715e34633 100644
--- a/src/test/java/io/getunleash/repository/UnleashEngineStateHandler.java
+++ b/src/test/java/io/getunleash/repository/UnleashEngineStateHandler.java
@@ -55,7 +55,7 @@ public MetricsBucket captureMetrics() {
     public void setState(String raw) {
         try {
             this.unleashEngine.takeState(raw);
-        } catch (YggdrasilInvalidInputException e) {
+        } catch (YggdrasilInvalidInputException | YggdrasilError e) {
             throw new RuntimeException(e);
         }
     }