Fix Go tests