Change tests for module to implement new interface