Move main_test.go to util_test.go