diff --git a/sdk/python/tests/unit/infra/test_streaming_ingestion.py b/sdk/python/tests/unit/infra/test_streaming_ingestion.py
index 259e16a3f5..74a39cefe0 100644
--- a/sdk/python/tests/unit/infra/test_streaming_ingestion.py
+++ b/sdk/python/tests/unit/infra/test_streaming_ingestion.py
@@ -14,6 +14,8 @@
 import assertpy
 import pytest
 
+from datetime import timedelta
+
 from feast.entity import Entity
 from feast.value_type import ValueType
 from feast.infra.contrib.spark_kafka_processor import SparkKafkaProcessor