diff --git a/train.py b/train.py
index 668c9162..b58b513a 100644
--- a/train.py
+++ b/train.py
@@ -121,7 +121,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
         mel_loss = criterion(mel_output, mel_spec, mel_lengths)
         linear_loss = criterion(linear_output, linear_spec, mel_lengths)
         if c.priority_freq:
-            linear_loss =  0.5 * linear_loss 
+            linear_loss =  0.5 * linear_loss\
                 + 0.5 * criterion(linear_output[:, :, :n_priority_freq],
                                   linear_spec[:, :, :n_priority_freq],
                                   mel_lengths)
@@ -248,7 +248,7 @@ def evaluate(model, criterion, data_loader, current_step):
             mel_loss = criterion(mel_output, mel_spec, mel_lengths)
             linear_loss = criterion(linear_output, linear_spec, mel_lengths)
             if c.priority_freq:
-                linear_loss =  0.5 * linear_loss 
+                linear_loss =  0.5 * linear_loss\
                     + 0.5 * criterion(linear_output[:, :, :n_priority_freq],
                                       linear_spec[:, :, :n_priority_freq],
                                       mel_lengths)