Cleaned up test for multi_columns