diff --git a/client_library/client_dbus.cc b/client_library/client_dbus.cc
index f16b759..4ec76c5 100644
--- a/client_library/client_dbus.cc
+++ b/client_library/client_dbus.cc
@@ -214,7 +214,7 @@
 
 bool DBusUpdateEngineClient::RegisterStatusUpdateHandler(
     StatusUpdateHandler* handler) {
-  if (!base::MessageLoopForIO::current()) {
+  if (!base::MessageLoopCurrent::IsSet()) {
     LOG(FATAL) << "Cannot get UpdateEngineClient outside of message loop.";
     return false;
   }