diff --git a/python/common.py b/python/common.py
index 7acfffe..966ff4b 100644
--- a/python/common.py
+++ b/python/common.py
@@ -60,11 +60,13 @@ def unit_test(TS: List[int], labels: List[str] = ['CPU', 'GPU', 'PY', 'PGPU'], t
 		title = f'X_{set_name}_feat_argsort'
 		print(f'Loading {title}...', file = stderr, end = '\r')
 		feat = None
+		#indices = pickle_multi_loader(['indices'], OUT_DIR)[0]
 		bs = []
 		for label in labels:
 			if feat is None:
 				feat_tmp = pickle_multi_loader([f'X_{set_name}_feat_{label}'], OUT_DIR)[0]
 				if feat_tmp is not None:
+					#feat = feat_tmp[indices]
 					feat = feat_tmp
 			bs.append(pickle_multi_loader([f'{title}_{label}'], OUT_DIR)[0])