diff --git a/main.py b/main.py
index 5de4b8d..a521fef 100644
--- a/main.py
+++ b/main.py
@@ -54,8 +54,8 @@ def main(args):
 
     if args.bert_embedder:
         bertEmbedder = BertGen(multilingualIndex, batch_size=args.batch_bert, nepochs=args.nepochs_bert,
+                               zero_shot=zero_shot, train_langs=zscl_train_langs,  # Todo: testing zero shot
                                patience=args.patience_bert, gpus=args.gpus, n_jobs=args.n_jobs)
-        bertEmbedder.transform(lX)
         embedder_list.append(bertEmbedder)
 
     # Init DocEmbedderList (i.e., first-tier learners or view generators) and metaclassifier