diff --git a/python/hsfs/engine/spark.py b/python/hsfs/engine/spark.py
index f20cc88b5..b3f64e87d 100644
--- a/python/hsfs/engine/spark.py
+++ b/python/hsfs/engine/spark.py
@@ -523,7 +523,7 @@ def save_stream_dataframe(
                 "wait_for_online_ingestion", False
             ):
                 feature_group.get_latest_online_ingestion().wait_for_completion(
-                    options=offline_write_options.get("online_ingestion_options", {})
+                    options=write_options.get("online_ingestion_options", {})
                 )
 
         return query
@@ -586,7 +586,7 @@ def _save_online_dataframe(self, feature_group, dataframe, write_options):
             "wait_for_online_ingestion", False
         ):
             feature_group.get_latest_online_ingestion().wait_for_completion(
-                options=offline_write_options.get("online_ingestion_options", {})
+                options=write_options.get("online_ingestion_options", {})
             )
 
     def _get_headers(