diff --git a/ulmfit/__main__.py b/ulmfit/__main__.py
index 23aec10..41c00ce 100644
--- a/ulmfit/__main__.py
+++ b/ulmfit/__main__.py
@@ -20,7 +20,7 @@ class ULMFiT:
     @wraps(CLSHyperParams)
     def cls(self, dataset_path, base_lm_path, **changes):
         params = CLSHyperParams.from_lm(dataset_path, base_lm_path, **changes)
-        return FireView(train=params.train_cls)
+        return FireView(train=params.train_cls, validate_cls=params.validate_cls)
 
 if __name__ == '__main__':
     fire.Fire(ULMFiT())
\ No newline at end of file