diff --git a/tests/test_offline.py b/tests/test_offline.py
index d34242bbd428360600767285a7a0c06fb0d931d7..b285338929f2ac03e51cf047cc6334dc217807f1 100644
--- a/tests/test_offline.py
+++ b/tests/test_offline.py
@@ -74,8 +74,8 @@ class TestOfflineKeys(unittest.TestCase):
         values = [i for i in range(18)]
 
         self.assertEqual(18, len([*fit.keys()]))
-        for k, v in zip(keys, values):
-            self.assertEqual(v, fit[k])
+        for k, v in fit.items():
+            self.assertEqual(values[v], fit[k])
 
 class TestReader(unittest.TestCase):
     def setUp(self):