diff --git a/includes/registration.entity.inc b/includes/registration.entity.inc
index 4892933..d72ced2 100644
--- a/includes/registration.entity.inc
+++ b/includes/registration.entity.inc
@@ -93,8 +93,8 @@ function registration_entity_info() {
 
   // Support entity cache module.
   if (module_exists('entitycache')) {
-    $return['registration']['field cache'] = FALSE;
-    $return['registration']['entity cache'] = TRUE;
+    $entities['registration']['field cache'] = FALSE;
+    $entities['registration']['entity cache'] = TRUE;
   }
 
   return $entities;