fix test_main again