Fixing bug showing the missing columns