fixrd issue in main_test.go