Index: blimp/net/blimp_message_checkpointer.cc |
diff --git a/blimp/net/blimp_message_checkpointer.cc b/blimp/net/blimp_message_checkpointer.cc |
index c9a20c37c6e4903e578d79fb90feac5d0304cafe..e1354da0cddd138c8be492492d4fc9dfda632777 100644 |
--- a/blimp/net/blimp_message_checkpointer.cc |
+++ b/blimp/net/blimp_message_checkpointer.cc |
@@ -35,9 +35,7 @@ BlimpMessageCheckpointer::~BlimpMessageCheckpointer() {} |
void BlimpMessageCheckpointer::ProcessMessage( |
std::unique_ptr<BlimpMessage> message, |
const net::CompletionCallback& callback) { |
- if (message->type() == BlimpMessage::PROTOCOL_CONTROL && |
- message->protocol_control().type() == |
- ProtocolControlMessage::CHECKPOINT_ACK) { |
+ if (message->has_protocol_control()) { |
if (message->protocol_control().has_checkpoint_ack() && |
message->protocol_control().checkpoint_ack().has_checkpoint_id()) { |
checkpoint_observer_->OnMessageCheckpoint( |