Fix assertion in checkout tests