Update to support new release of checkout module