diff --git a/vall_e/config.py b/vall_e/config.py
index 958361e..3ff8a66 100755
--- a/vall_e/config.py
+++ b/vall_e/config.py
@@ -475,8 +475,7 @@ try:
 	if not cfg.dataset.use_hdf5:
 		cfg.dataset.training = [ Path(dir) for dir in cfg.dataset.training ]
 		cfg.dataset.validation = [ Path(dir) for dir in cfg.dataset.validation ]
-	
-	cfg.dataset.noise = [ Path(dir) for dir in cfg.dataset.noise ]
+		cfg.dataset.noise = [ Path(dir) for dir in cfg.dataset.noise ]
 except Exception as e:
 	pass
 
diff --git a/vall_e/data.py b/vall_e/data.py
index d746e2f..e455723 100755
--- a/vall_e/data.py
+++ b/vall_e/data.py
@@ -638,6 +638,10 @@ def create_dataset_hdf5():
 	for data_dir in tqdm(cfg.dataset.validation, desc='Processing Validation'):
 		add( data_dir, type="validation" )
 
+	# noise
+	for data_dir in tqdm(cfg.dataset.noise, desc='Processing Noise'):
+		add( data_dir, type="noise" )
+
 	# write symmap
 	hf.create_dataset('symmap', data=json.dumps(symmap))