Set remove_unused_columns=False to prevent this behavior!