Code download does not properly work for UI flow