diff --git a/src/main/java/com/bynder/sdk/query/upload/SaveMediaQuery.java b/src/main/java/com/bynder/sdk/query/upload/SaveMediaQuery.java
index 5e6cfc2..b4e700e 100644
--- a/src/main/java/com/bynder/sdk/query/upload/SaveMediaQuery.java
+++ b/src/main/java/com/bynder/sdk/query/upload/SaveMediaQuery.java
@@ -102,7 +102,7 @@ public SaveMediaQuery setMetaproperties(List<MetapropertyAttribute> metaproperti
         return this;
     }
 
-    public SaveMediaQuery setIsPublic(final boolean isPublic) {
+    public SaveMediaQuery setIsPublic(final Boolean isPublic) {
         this.isPublic = isPublic;
         return this;
     }