diff --git a/relation.module b/relation.module
index 9dab6c6..51dbb3d 100644
--- a/relation.module
+++ b/relation.module
@@ -97,7 +97,7 @@ function relation_entity_property_info_alter(&$info) {
   foreach (relation_get_types() as $relation_type => $relation) {
     foreach ($relation->source_bundles as $key => $bundles) {
       list($entity_type, $bundle) = explode(':', $bundles, 2);
-      $info['relation']['bundles'][$bundle]['properties']['endpoints_source_' . $entity_type] = array(
+      $info['relation']['bundles'][$relation_type]['properties']['endpoints_source_' . $entity_type] = array(
         'label' => t('@type (source endpoint)', array('@type' => $entity_type)),
         'type' => 'list<' . $entity_type . '>',
         'getter callback' => 'relation_rules_get_specific_endpoints',
@@ -108,7 +108,7 @@ function relation_entity_property_info_alter(&$info) {
 
     foreach ($relation->target_bundles as $key => $bundles) {
       list($entity_type, $bundle) = explode(':', $bundles, 2);
-      $info['relation']['bundles'][$bundle]['properties']['endpoints_target_' . $entity_type] = array(
+      $info['relation']['bundles'][$relation_type]['properties']['endpoints_target_' . $entity_type] = array(
         'label' => t('@type (target endpoint)', array('@type' => $entity_type)),
         'type' => 'list<' . $entity_type . '>',
         'getter callback' => 'relation_rules_get_specific_endpoints',