Can you add tests for your changes in this module?