modifying tests to match the module