diff --git a/sedkit/tests/test_modelgrid.py b/sedkit/tests/test_modelgrid.py
index 2b88f17..464bc05 100644
--- a/sedkit/tests/test_modelgrid.py
+++ b/sedkit/tests/test_modelgrid.py
@@ -57,7 +57,7 @@ def test_filter(self):
     def test_photometry(self):
         """Test the photometry method"""
         phot_mg = self.bt.photometry(['2MASS.J', '2MASS.H', '2MASS.Ks'])
-        self.assertEqual(phot_mg.index.iloc[0]['spectrum'][0].size, 3)
+        self.assertEqual(phot_mg.index['spectrum'][0][0].size, 3)
 
     def test_plot(self):
         """Test that the plot method works"""