Fix connection of flat field correction