For this reason, set remove_unused_columns to False.