Minor fix for Go