Fixed syntax error in unit.go