diff --git a/graphstorm-processing/graphstorm_processing/config/config_parser.py b/graphstorm-processing/graphstorm_processing/config/config_parser.py
index b94d4b9e6c..ee9abac20f 100644
--- a/graphstorm-processing/graphstorm_processing/config/config_parser.py
+++ b/graphstorm-processing/graphstorm_processing/config/config_parser.py
@@ -22,8 +22,7 @@
 from .label_config_base import LabelConfig, EdgeLabelConfig, NodeLabelConfig
 from .feature_config_base import FeatureConfig, NoopFeatureConfig
 from .numerical_configs import (MultiNumericalFeatureConfig,
-                                NumericalFeatureConfig,
-                                BucketFeatureConfig)
+                                NumericalFeatureConfig)
 from .data_config_base import DataStorageConfig