Corrected the missing columns issue