diff --git a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py
index 0461ea1b..f5adce62 100644
--- a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py
+++ b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py
@@ -79,7 +79,7 @@ class PreprocessorReference(Preprocessor):
                 return h
 
             if self.is_recording_style:
-                a = 0
+                self.recorded_h[location] = h
             else:
                 b = 0
 
@@ -97,7 +97,7 @@ class PreprocessorReference(Preprocessor):
                         transformer_options['block_index'])
 
             if self.is_recording_style:
-                a = 0
+                self.recorded_attn1[location] = (k, v)
             else:
                 b = 0