diff --git a/mock_system_state.h b/mock_system_state.h
index de0caef..e18c403 100644
--- a/mock_system_state.h
+++ b/mock_system_state.h
@@ -15,7 +15,7 @@
 #include "update_engine/mock_p2p_manager.h"
 #include "update_engine/mock_payload_state.h"
 #include "update_engine/clock.h"
-#include "update_engine/hardware.h"
+#include "update_engine/fake_hardware.h"
 #include "update_engine/prefs_mock.h"
 #include "update_engine/system_state.h"
 
@@ -97,6 +97,10 @@
     hardware_ = hardware;
   }
 
+  inline FakeHardware* get_fake_hardware() {
+    return &default_hardware_;
+  }
+
   inline void set_prefs(PrefsInterface* prefs) {
     prefs_ = prefs;
   }
@@ -142,7 +146,7 @@
 
   // These are the other object we own.
   Clock default_clock_;
-  Hardware default_hardware_;
+  FakeHardware default_hardware_;
   OmahaRequestParams default_request_params_;
 
   // These are pointers to objects which caller can override.