diff --git a/entity.features.inc b/entity.features.inc
index e266aaf..1cbe786 100644
--- a/entity.features.inc
+++ b/entity.features.inc
@@ -133,7 +133,7 @@ class EntityDefaultFeaturesController {
 }
 
 /**
- * Implements of hook_features_api().
+ * Implements hook_features_api().
  */
 function entity_features_api() {
   $items = array();
@@ -146,7 +146,7 @@ function entity_features_api() {
 }
 
 /**
- * Features component callback.
+ * Implements hook_features_export_options().
  */
 function entity_features_export_options($a1, $a2 = NULL) {
   // Due to a change in the Features API the first parameter might be a feature
@@ -157,22 +157,30 @@ function entity_features_export_options($a1, $a2 = NULL) {
 }
 
 /**
- * Features component callback.
+ * Implements hook_features_export().
  */
 function entity_features_export($data, &$export, $module_name = '', $entity_type) {
   return entity_features_get_controller($entity_type)->export($data, $export, $module_name);
 }
 
 /**
- * Features component callback.
+ * Implements hook_features_export_render().
  */
 function entity_features_export_render($module, $data, $export = NULL, $entity_type) {
   return entity_features_get_controller($entity_type)->export_render($module, $data, $export);
 }
 
 /**
- * Features component callback.
+ * Implements hook_features_revert().
  */
 function entity_features_revert($module = NULL, $entity_type) {
-  return entity_features_get_controller($entity_type)->revert($module);
+  entity_features_get_controller($entity_type)->revert($module);
+  entity_defaults_rebuild(array($entity_type));
+}
+
+/**
+ * Implements hook_features_rebuild().
+ */
+function entity_features_rebuild($module = NULL, $entity_type) {
+  entity_defaults_rebuild(array($entity_type));
 }