Improve test module