fixed test for dataframe