Fixed module name in test_main.