diff --git a/tests/test_tacotron2_model.py b/tests/test_tacotron2_model.py
index c2c39231..bda83a03 100644
--- a/tests/test_tacotron2_model.py
+++ b/tests/test_tacotron2_model.py
@@ -38,7 +38,7 @@ class TacotronTrainTest(unittest.TestCase):
 
         criterion = MSELossMasked().to(device)
         criterion_st = nn.BCEWithLogitsLoss().to(device)
-        model = Tacotron2(24, c.r, 5).to(device)
+        model = Tacotron2(num_chars=24, r=c.r, num_speakers=5).to(device)
         model.train()
         model_ref = copy.deepcopy(model)
         count = 0
diff --git a/tests/test_tacotron_model.py b/tests/test_tacotron_model.py
index a80b2080..7b343423 100644
--- a/tests/test_tacotron_model.py
+++ b/tests/test_tacotron_model.py
@@ -45,8 +45,8 @@ class TacotronTrainTest(unittest.TestCase):
         criterion = L1LossMasked().to(device)
         criterion_st = nn.BCEWithLogitsLoss().to(device)
         model = Tacotron(
-            32,
-            5, 
+            num_chars=32,
+            num_speakers=5, 
             linear_dim=c.audio['num_freq'],
             mel_dim=c.audio['num_mels'],
             r=c.r,