Ensured that custom_columns flags are being passed correctly.