diff --git a/pvliveconsumer/nightime.py b/pvliveconsumer/nightime.py
index 3753e21..af849d4 100644
--- a/pvliveconsumer/nightime.py
+++ b/pvliveconsumer/nightime.py
@@ -1,8 +1,7 @@
 """ Add zeros to nightime """
 import logging
 import os
-from datetime import datetime
-
+from datetime import datetime, timedelta
 import pandas as pd
 import pvlib
 from nowcasting_datamodel.models.gsp import LocationSQL