diff --git a/chebai/cli.py b/chebai/cli.py
index 36245aa..b7e78d1 100644
--- a/chebai/cli.py
+++ b/chebai/cli.py
@@ -47,13 +47,6 @@ def add_arguments_to_parser(self, parser: LightningArgumentParser):
         parser.link_arguments(
             "model.init_args.out_dim", "trainer.callbacks.init_args.num_labels"
         )
-        parser.link_arguments(
-            "data", "model.init_args.criterion.init_args.data_extractor"
-        )
-        parser.link_arguments(
-            "data.init_args.chebi_version",
-            "model.init_args.criterion.init_args.data_extractor.init_args.chebi_version",
-        )
 
     @staticmethod
     def subcommands() -> Dict[str, Set[str]]: