Improve module test case