[Fix] ScheduleFreeWrapper

[kozistr]

Problem (Why?)

Close #363

Solution (What/How?)

• initialize z state within step()

Other changes (bug fixes, small refactors)

N/A

Notes

N/A

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 100.00%. Comparing base (aca2ef2) to head (9efb597).
Report is 6 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff            @@
##              main      #367   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files          113       113           
  Lines         9243      9241    -2     
=========================================
- Hits          9243      9241    -2     

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

🚀 New features to boost your workflow:

• ❄ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.