diff --git a/relation.module b/relation.module
index 2b6c15e..ce741ad 100644
--- a/relation.module
+++ b/relation.module
@@ -960,9 +960,9 @@ function relation_get_available_types($entity_type, $bundle, $endpoint = 'source
 }
 
 /**
- * Implements hook_entity_delete().
+ * Implements hook_field_attach_delete().
  */
-function relation_entity_delete($entity, $entity_type) {
+function relation_field_attach_delete($entity_type, $entity) {
   list($entity_id) = entity_extract_ids($entity_type, $entity);
   $relations = relation_query($entity_type, $entity_id)->execute();
   $relations_to_delete = array();