diff --git a/MARBLE/main.py b/MARBLE/main.py
index 4f58647d..9a52c461 100644
--- a/MARBLE/main.py
+++ b/MARBLE/main.py
@@ -378,9 +378,7 @@ def fit(self, data, outdir=None, verbose=False):
             train_loss, optimizer = self.batch_loss(
                 data, train_loader, train=True, verbose=verbose, optimizer=optimizer
             )
-            val_loss, _ = self.batch_loss(
-                data, val_loader, verbose=verbose
-            )
+            val_loss, _ = self.batch_loss(data, val_loader, verbose=verbose)
             scheduler.step(train_loss)
 
             print(