Make Go code work with test_locally.sh