Fixes return type of remove_entity_relationship