diff --git a/sites/all/modules/contrib/features/features.module b/sites/all/modules/contrib/features/features.module
index 5c30610..a05b1dd 100755
--- a/sites/all/modules/contrib/features/features.module
+++ b/sites/all/modules/contrib/features/features.module
@@ -685,7 +685,7 @@ function _features_restore($op, $items = array()) {
   }
 
   if (empty($items)) {
-    $states = features_get_component_states();
+    $states = features_get_component_states(array(), ($op == 'rebuild'));
     foreach ($states as $module_name => $components) {
       foreach ($components as $component => $state) {
         if (in_array($state, $restore_states)) {