Move tests to a module