Fix order#update_shipment_state logic