diff --git a/tests/src/python/provider_python.py b/tests/src/python/provider_python.py
index cd338be86589..a6a06197a8c2 100644
--- a/tests/src/python/provider_python.py
+++ b/tests/src/python/provider_python.py
@@ -92,7 +92,7 @@ def fetchFeature(self, f):
                 _f = self._source._features[list(self._source._features.keys())[self._index]]
                 self._index += 1
 
-                if self._feature_id_list is not None and f.id() not in self._feature_id_list:
+                if self._feature_id_list is not None and _f.id() not in self._feature_id_list:
                     continue
 
                 if not self._filter_rect.isNull():
@@ -387,7 +387,7 @@ def supportsSubsetString(self):
     def createSpatialIndex(self):
         if self._spatialindex is None:
             self._spatialindex = QgsSpatialIndex()
-            for f in self._features:
+            for f in self._features.values():
                 self._spatialindex.insertFeature(f)
         return True