diff --git a/distribute.py b/distribute.py
index a5fdb373..873d8aba 100644
--- a/distribute.py
+++ b/distribute.py
@@ -9,7 +9,8 @@ import torch.distributed as dist
 from torch.utils.data.sampler import Sampler
 from torch.autograd import Variable
 from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
-from TTS.utils.generic_utils import load_config, create_experiment_folder
+from TTS.utils.io import load_config
+from TTS.utils.generic_utils import create_experiment_folder
 
 
 class DistributedSampler(Sampler):