diff --git a/integrations/keras/prune_resnet20.py b/integrations/keras/prune_resnet20.py
index 89b468d48fd..2dfb12f1ea3 100644
--- a/integrations/keras/prune_resnet20.py
+++ b/integrations/keras/prune_resnet20.py
@@ -77,7 +77,7 @@ def download_model_and_recipe(root_dir: str):
         dataset="cifar_10",
         training_scheme=None,
         optim_name="base",
-        optim_category=None,
+        optim_category="none",
         optim_target=None,
         override_parent_path=model_dir,
     )