diff --git a/test/feature/test_holiday.py b/test/feature/test_holiday.py
index 582901f..6c582bb 100644
--- a/test/feature/test_holiday.py
+++ b/test/feature/test_holiday.py
@@ -41,7 +41,9 @@ from pts.feature.holiday import (
     CYBER_MONDAY,
     SpecialDateFeatureSet,
     squared_exponential_kernel,
-    CustomDateFeatureSet
+    exponential_kernel,
+    CustomDateFeatureSet,
+    CustomHolidayFeatureSet
 )
 
 test_dates = {