Fix go tests