Fix to previous update