fix test for dataframe