diff --git a/openpilot/sunnypilot/modeld_v2/compile_modeld.py b/openpilot/sunnypilot/modeld_v2/compile_modeld.py
index 7959aa9e5c..aee8984bce 100755
--- a/openpilot/sunnypilot/modeld_v2/compile_modeld.py
+++ b/openpilot/sunnypilot/modeld_v2/compile_modeld.py
@@ -202,8 +202,8 @@ def make_run_policy(vision_runner, policy_runners: list, features_slice: slice,
 
     if 'prev_feat' in unpacked_dict:
       prev_feat_dev = unpacked_dict['prev_feat']
-      feat_buf = shift_and_sample(feat_q, prev_feat_dev.reshape(1, 1, -1), sample_skip_fn).realize()
-      inputs['features_buffer'] = feat_buf if len(fb := input_shapes['features_buffer']) <= 3 else feat_buf.reshape(fb)
+      feat_buf = shift_and_sample(feat_q, prev_feat_dev.reshape(1, 1, -1), sample_skip_fn)
+      inputs['features_buffer'] = (feat_buf if len(fb := input_shapes['features_buffer']) <= 3 else feat_buf.reshape(fb)).realize()
 
     if vision_runner:
       vision_out_cast = next(iter(vision_runner({road_key: img, wide_key: big_img}).values())).cast('float32').realize()
@@ -216,7 +216,7 @@ def make_run_policy(vision_runner, policy_runners: list, features_slice: slice,
     inputs.update({road_key: img, wide_key: big_img})
     if 'features_buffer' not in inputs:
       feat_buf = sample_skip_fn(feat_q)
-      inputs['features_buffer'] = feat_buf.reshape(input_shapes['features_buffer'])
+      inputs['features_buffer'] = (feat_buf if len(fb := input_shapes.get('features_buffer', ())) <= 3 else feat_buf.reshape(fb)).realize()
 
     policy_out = next(iter(policy_runners[0](inputs).values())).cast('float32').realize()
     if 'features_buffer' not in inputs and features_slice is not None: