fix test_main