Fix failing CI tests on main

[daico007]

PR Summary:

Update CI.yml to fix the failing tests, address #550.

PR Checklist

---

• Includes appropriate unit test(s)
• Appropriate docstring(s) are added/updated
• Code is (approximately) PEP8 compliant
• Issue(s) raised/addressed?

[codecov]

Codecov Report

Merging #551 (908b645) into main (d2608b7) will decrease coverage by 7.43%.
Report is 4 commits behind head on main.
The diff coverage is n/a.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #551      +/-   ##
==========================================
- Coverage   69.21%   61.79%   -7.43%     
==========================================
  Files          16       16              
  Lines        1673     1662      -11     
==========================================
- Hits         1158     1027     -131     
- Misses        515      635     +120