## Description - The test coverage in the codebase is around 49%. Need to add a more balanced test case to maintain high test coverage