Move tests in a separate module