diff --git a/update_engine.gyp b/update_engine.gyp
index 20fb282..cbdce2b 100644
--- a/update_engine.gyp
+++ b/update_engine.gyp
@@ -293,6 +293,7 @@
         'update_manager/real_system_provider.cc',
         'update_manager/real_time_provider.cc',
         'update_manager/real_updater_provider.cc',
+        'update_manager/staging_utils.cc',
         'update_manager/state_factory.cc',
         'update_manager/update_manager.cc',
         'update_manager/update_time_restrictions_policy_impl.cc',
@@ -581,6 +582,7 @@
             'update_manager/real_system_provider_unittest.cc',
             'update_manager/real_time_provider_unittest.cc',
             'update_manager/real_updater_provider_unittest.cc',
+            'update_manager/staging_utils_unittest.cc',
             'update_manager/umtest_utils.cc',
             'update_manager/update_manager_unittest.cc',
             'update_manager/update_time_restrictions_policy_impl_unittest.cc',