fix #1483

[remicres]

Related Issue(s):

• Classification extension not working with classes #1483

Description:

Quick fix for #1483 1483

PR Checklist:

• Pre-commit hooks and tests pass (run scripts/test)
• Documentation has been updated to reflect changes, if applicable
• This PR maintains or improves overall codebase code coverage.
• Changes are added to the CHANGELOG. See the docs for information about adding to the changelog.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 91.17%. Comparing base (97f87cb) to head (d8e5645).
Report is 1 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1484   +/-   ##
=======================================
  Coverage   91.17%   91.17%           
=======================================
  Files          52       52           
  Lines        7251     7253    +2     
  Branches      875      877    +2     
=======================================
+ Hits         6611     6613    +2     
  Misses        460      460           
  Partials      180      180           

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

[jsignell]

This looks good to me! If you are able to add a test that would be great.

[remicres]

Ok I'll try to add a test