fix: unit tests

[athith-g]

Description

Fixes #214.

Type of change

• [x] Bug fix (non-breaking change which fixes an issue)

Checklist:

• [x] My code follows the style guidelines of this project
• [x] I have performed a self-review of my own code
• [x] I have updated the documentation (or documentation does not need to be updated)
• [x] My changes generate no new warnings
• [x] New and existing unit tests pass locally with my changes
• [x] I have not reduced the existing code coverage