Improve test coverage

[satvik-s]

Type of Issue

• Bug
• Feature
• Docs/Chore
• Support

Issue Description

Improve test coverage for the kotlin implementation

Steps to Reproduce the Problem (applicable for a bug)

• N/A

Specifications

• Version: N/A
• Platform: N/A