Fix test faling due to DataFrames changes.