1 files changed, 4 insertions, 2 deletions
diff --git a/chromeos/disks/disk_mount_manager_unittest.cc b/chromeos/disks/disk_mount_manager_unittest.cc
index 83def45..36e67ad 100644
--- a/chromeos/disks/disk_mount_manager_unittest.cc
+++ b/chromeos/disks/disk_mount_manager_unittest.cc
@@ -122,9 +122,11 @@ class DiskMountManagerTest : public testing::Test {
   // Adds a test observer to the disk mount manager.
   virtual void SetUp() {
     FakeDBusThreadManager* fake_thread_manager = new FakeDBusThreadManager();
-    DBusThreadManager::InitializeForTesting(fake_thread_manager);
+    fake_cros_disks_client_ = new FakeCrosDisksClient;
+    fake_thread_manager->SetCrosDisksClient(
+        scoped_ptr<CrosDisksClient>(fake_cros_disks_client_));
 
-    fake_cros_disks_client_ = fake_thread_manager->fake_cros_disks_client();
+    DBusThreadManager::InitializeForTesting(fake_thread_manager);
 
     DiskMountManager::Initialize();