Fix breaking doc tests

[aronisstav]

Summary

Fix breaking doc tests

Checklist

• Has tests (or doesn't need them)
• Updates CHANGELOG (or too minor)
• References related Issues

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 95.45%. Comparing base (cbc5200) to head (57d9bde).
Report is 4 commits behind head on master.

Additional details and impacted files

@@           Coverage Diff           @@
##           master     #358   +/-   ##
=======================================
  Coverage   95.45%   95.45%           
=======================================
  Files          12       12           
  Lines        3738     3738           
=======================================
  Hits         3568     3568           
  Misses        170      170           

Flag	Coverage Δ
tests-1	76.99% <ø> (ø)
tests-2	68.78% <ø> (ø)
tests-real	84.80% <ø> (ø)
tests-unit	2.65% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

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