diff --git a/adv_control/control_svd.py b/adv_control/control_svd.py
index 9458f6a..6a7de04 100644
--- a/adv_control/control_svd.py
+++ b/adv_control/control_svd.py
@@ -305,7 +305,7 @@ class SVDControlNet(nn.Module):
 
         cond = kwargs["cond"]
         num_video_frames = cond["num_video_frames"]
-        image_only_indicator = cond.get("image_only_indicator", None)
+        image_only_indicator = cond["image_only_indicator"]
         time_context = cond.get("time_context", None)
         del cond