Update protocol.go