diff --git a/recbole/data/dataset/dataset.py b/recbole/data/dataset/dataset.py
index 754f32461..5d8b6a6f9 100644
--- a/recbole/data/dataset/dataset.py
+++ b/recbole/data/dataset/dataset.py
@@ -1474,6 +1474,7 @@ def build(self):
         self._change_feat_format()
 
         if self.benchmark_filename_list is not None:
+            self._drop_unused_col()
             cumsum = list(np.cumsum(self.file_size_list))
             datasets = [self.copy(self.inter_feat[start:end]) for start, end in zip([0] + cumsum[:-1], cumsum)]
             return datasets