diff --git a/.travis/script b/.travis/script
index 0c24a221..0860f9cf 100755
--- a/.travis/script
+++ b/.travis/script
@@ -17,5 +17,6 @@ fi
 if [[ "$TEST_SUITE" == "testscripts" ]]; then
    # test model training scripts
   ./tests/test_tts_train.sh
-  ./tests/test_vocoder_train.sh
+  ./tests/test_vocoder_gan_train.sh
+  ./tests/test_vocoder_wavernn_train.sh
 fi
diff --git a/TTS/vocoder/models/wavernn.py b/TTS/vocoder/models/wavernn.py
index 9b151cac..8a45d9e3 100644
--- a/TTS/vocoder/models/wavernn.py
+++ b/TTS/vocoder/models/wavernn.py
@@ -225,7 +225,7 @@ class WaveRNN(nn.Module):
 
     def forward(self, x, mels):
         bsize = x.size(0)
-        h1 = torch.zeros(1, bsize, self.rnn_dims).to(x.device) 
+        h1 = torch.zeros(1, bsize, self.rnn_dims).to(x.device)
         h2 = torch.zeros(1, bsize, self.rnn_dims).to(x.device)
         mels, aux = self.upsample(mels)