diff --git a/tests/test_schema.py b/tests/test_schema.py
index 6d35fc8..11643d3 100644
--- a/tests/test_schema.py
+++ b/tests/test_schema.py
@@ -50,9 +50,8 @@ def _validateSchema(self, dataset, dataId, tableName):
         self.assertEqual(len(sdmSchema), 1)
         expectedColumnNames = set(column['name'] for column in sdmSchema[0]['columns'])
 
-        df = self.butler.get(dataset, dataId)
-        df.reset_index(inplace=True)
-        outputColumnNames = set(df.columns.to_list())
+        df = self.butler.get(dataset, dataId, storageClass="ArrowAstropy")
+        outputColumnNames = set(df.colnames)
         self.assertEqual(outputColumnNames, expectedColumnNames)
 
     def testObjectSchemaMatch(self):