diff --git a/message.module b/message.module
index 7708c94..f5e5811 100644
--- a/message.module
+++ b/message.module
@@ -195,7 +195,7 @@ function message_entity_info() {
     ),
     'bundles' => array(),
     'bundle keys' => array(
-      'bundle' => 'name',
+      'bundle' => 'type',
     ),
     'view modes' => array(
       'full' => array(