Update protocol_test.go