diff --git a/main.py b/main.py
index 9e41351..583fff8 100644
--- a/main.py
+++ b/main.py
@@ -18,6 +18,7 @@
 import csv
 import pandas as pd
 from datetime import datetime
+from datetime import timedelta
 import os
 from azure.cosmos.exceptions import CosmosResourceExistsError
 import azure.cosmos.cosmos_client as cosmos_client