diff --git a/rasa_core/train.py b/rasa_core/train.py
index 8c473dfec12..e9fbb7d076c 100644
--- a/rasa_core/train.py
+++ b/rasa_core/train.py
@@ -227,7 +227,10 @@ def train_dialogue_model(domain_file, stories_file, output_path,
         else:
             logger.info("loading a pre-trained model. ",
                         "all training-related parameters will be ignored")
-        _agent = Agent.load(cmdline_args.core, interpreter=_interpreter)
+        _agent = Agent.load(cmdline_args.core,
+                            interpreter=_interpreter,
+                            generator=_endpoints.nlg,
+                            action_endpoint=_endpoints.action)
     else:
         if not cmdline_args.out:
             raise ValueError("you must provide a path where the model "