diff --git a/server/synthesizer.py b/server/synthesizer.py
index c0069e33..5eb7f0d4 100644
--- a/server/synthesizer.py
+++ b/server/synthesizer.py
@@ -7,7 +7,8 @@ import torch
 import yaml
 
 from TTS.utils.audio import AudioProcessor
-from TTS.utils.generic_utils import load_config, setup_model
+from TTS.utils.io import load_config
+from TTS.utils.generic_utils import setup_model
 from TTS.utils.speakers import load_speaker_mapping
 # pylint: disable=unused-wildcard-import
 # pylint: disable=wildcard-import
diff --git a/tests/test_audio.py b/tests/test_audio.py
index 2ede77ce..4b8ee276 100644
--- a/tests/test_audio.py
+++ b/tests/test_audio.py
@@ -3,7 +3,7 @@ import unittest
 
 from TTS.tests import get_tests_path, get_tests_input_path, get_tests_output_path
 from TTS.utils.audio import AudioProcessor
-from TTS.utils.generic_utils import load_config
+from TTS.utils.io import load_config
 
 TESTS_PATH = get_tests_path()
 OUT_PATH = os.path.join(get_tests_output_path(), "audio_tests")
@@ -172,4 +172,4 @@ class TestAudio(unittest.TestCase):
         mel_reference = self.ap.melspectrogram(wav)
         mel_norm = ap.melspectrogram(wav)
         mel_denorm = ap._denormalize(mel_norm)
-        assert abs(mel_reference - mel_denorm).max() < 1e-4
\ No newline at end of file
+        assert abs(mel_reference - mel_denorm).max() < 1e-4
diff --git a/tests/test_demo_server.py b/tests/test_demo_server.py
index 11d16a45..65653997 100644
--- a/tests/test_demo_server.py
+++ b/tests/test_demo_server.py
@@ -22,7 +22,7 @@ class DemoServerTest(unittest.TestCase):
         num_chars = len(phonemes) if config.use_phonemes else len(symbols)
         model = setup_model(num_chars, 0, config)
         output_path = os.path.join(get_tests_output_path())
-        save_checkpoint(model, None, None, None, output_path, 10, 10)
+        save_checkpoint(model, None, None, None, 1, output_path)
 
     def test_in_out(self):
         self._create_random_model()
diff --git a/tests/test_tacotron2_tf_model.py b/tests/test_tacotron2_tf_model.py
index bc8f0407..aca363a8 100644
--- a/tests/test_tacotron2_tf_model.py
+++ b/tests/test_tacotron2_tf_model.py
@@ -60,4 +60,4 @@ class TacotronTFTrainTest(unittest.TestCase):
         assert output[3].shape[1] == (mel_spec.shape[1] // model.decoder.r)
 
         # inference pass
-        output = model(input, training=False)
+        output = model(chars_seq, training=False)