diff --git a/pts/feature/__init__.py b/pts/feature/__init__.py
index d1b5434..52fb5da 100644
--- a/pts/feature/__init__.py
+++ b/pts/feature/__init__.py
@@ -35,4 +35,4 @@ from pts.feature.transform import (
     SelectFields,
 )
 
-from pts.feature.lag import time_features_from_frequency_str, get_lags_for_frequency, get_granularity
+from pts.feature.lag import time_features_from_frequency_str, get_lags_for_frequency