diff --git a/examples/shields/rl/utils.py b/examples/shields/rl/utils.py
index 42560e9..c70e9b2 100644
--- a/examples/shields/rl/utils.py
+++ b/examples/shields/rl/utils.py
@@ -116,7 +116,7 @@ class MiniGridShieldHandler(ShieldHandler):
         state_valuations = model.state_valuations
         choice_labeling = model.choice_labeling
 
-        #stormpy.shields.export_shield(model, shield, self.tmp_dir_name + "/current.shield")
+        stormpy.shields.export_shield(model, shield, self.tmp_dir_name + "/shield")
 
         print(f"LOG: Starting to translate shield...")
         tic()