Fix test fails

Please provide a proper Issue summary

@dhruv.mittal: Thanks, but I'm the maintainer of this project, this is just a working issue so I can create a fork and merge request.

Automatically closed - issue fixed for 2 weeks with no activity.