Solve visualization bugs

[Tortar]

This solves the bug, which was actually introduced in #863, now everything works

edit: in the end, I will solve all visualization bugs I find when executing the examples

[codecov-commenter]

Codecov Report

Merging #867 (2333ac7) into main (5e70cc5) will increase coverage by 0.05%. The diff coverage is n/a.

:exclamation: Current head 2333ac7 differs from pull request most recent head 906ec9e. Consider uploading reports for the commit 906ec9e to get more accurate results

@@            Coverage Diff             @@
##             main     #867      +/-   ##
==========================================
+ Coverage   70.89%   70.95%   +0.05%     
==========================================
  Files          42       43       +1     
  Lines        2797     2799       +2     
==========================================
+ Hits         1983     1986       +3     
+ Misses        814      813       -1     

see 2 files with indirect coverage changes

:mega: We’re building smart automated test selection to slash your CI/CD build times. Learn more

[Tortar]

these changes solved all bugs, I managed to run all the examples one by one without any problem!