Add check for relation