Fix after change to amount