Update this test to use new dataframe outputs