This issue aims to improve the test coverage for this folder by:

1. Identifying untested or under-tested functions and classes.
2. Writing additional unit tests to cover edge cases and scenarios.