diff --git a/pystore/utils.py b/pystore/utils.py
index f58f747..4293a9f 100644
--- a/pystore/utils.py
+++ b/pystore/utils.py
@@ -19,7 +19,7 @@
 # limitations under the License.
 
 import os
-from datetime import datetime
+from datetime import datetime, timezone
 import json
 import shutil
 import pandas as pd