Tests: Refactor the main module test