Use relationship attribute instead of id