diff --git a/mock_connection_manager.h b/mock_connection_manager.h
index 522e1f6..c74d6ad 100644
--- a/mock_connection_manager.h
+++ b/mock_connection_manager.h
@@ -19,7 +19,7 @@
   MockConnectionManager(SystemState* system_state)
       : ConnectionManager(system_state) {}
 
-  MOCK_CONST_METHOD2(GetConnectionType, bool(DbusGlibInterface* dbus_iface,
+  MOCK_CONST_METHOD2(GetConnectionType, bool(DBusWrapperInterface* dbus_iface,
                                              NetworkConnectionType* out_type));
 
   MOCK_CONST_METHOD1(IsUpdateAllowedOver, bool(NetworkConnectionType type));