fix: Rework Truth Output

[HCookie]

Description

Override write_step instead of write_state.

Type of Change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• Documentation update

Code Compatibility

• I have performed a self-review of my code

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 96.07%. Comparing base (90728d5) to head (aa3272f).
Report is 73 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #159      +/-   ##
==========================================
- Coverage   98.03%   96.07%   -1.97%     
==========================================
  Files           3        3              
  Lines          51       51              
==========================================
- Hits           50       49       -1     
- Misses          1        2       +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.