chore: fix tests

[mdjastrzebski]

Summary

Test plan

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 95.56%. Comparing base (16f9311) to head (0f917c5).
Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1747      +/-   ##
==========================================
+ Coverage   95.52%   95.56%   +0.03%     
==========================================
  Files          93       93              
  Lines        5205     5205              
  Branches      879      881       +2     
==========================================
+ Hits         4972     4974       +2     
+ Misses        233      231       -2     

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