fixing indent error

[annahedstroem]

Description

• Fixing indent error, only looping through the last layer.

Implemented changes

• See three files

Minimum acceptance criteria

• Specify what is necessary for the PR to be merged with the main branch.
• @mentions of the person that is apt to review these changes e.g., @annahedstroem

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (c7d2012) 91.24% compared to head (99b6671) 91.24%.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #316   +/-   ##
=======================================
  Coverage   91.24%   91.24%           
=======================================
  Files          66       66           
  Lines        3884     3884           
=======================================
  Hits         3544     3544           
  Misses        340      340           

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