diff --git a/fake_system_state.cc b/fake_system_state.cc
index 8291ba3..8c5ffc9 100644
--- a/fake_system_state.cc
+++ b/fake_system_state.cc
@@ -3,9 +3,9 @@
 // found in the LICENSE file.
 
 #include "update_engine/fake_system_state.h"
-#include "update_engine/policy_manager/fake_state.h"
+#include "update_engine/update_manager/fake_state.h"
 
-using chromeos_policy_manager::FakeState;
+using chromeos_update_manager::FakeState;
 
 namespace chromeos_update_engine {
 
@@ -15,7 +15,7 @@
   : mock_connection_manager_(this),
     mock_update_attempter_(this, &dbus_),
     default_request_params_(this),
-    fake_policy_manager_(&fake_clock_),
+    fake_update_manager_(&fake_clock_),
     clock_(&fake_clock_),
     connection_manager_(&mock_connection_manager_),
     hardware_(&fake_hardware_),
@@ -26,7 +26,7 @@
     update_attempter_(&mock_update_attempter_),
     request_params_(&default_request_params_),
     p2p_manager_(&mock_p2p_manager_),
-    policy_manager_(&fake_policy_manager_),
+    update_manager_(&fake_update_manager_),
     device_policy_(nullptr),
     fake_system_rebooted_(false) {
   mock_payload_state_.Initialize(this);