Fix to use respond_to