Fix update quantity method