Add test coverage for the module