diff --git a/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java b/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java index 62de34961e..e541bef1d1 100644 --- a/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java +++ b/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java @@ -647,7 +647,6 @@ public Collection createComponents( Path dataPath = environment.dataFiles()[0]; mlFeatureEnabledSetting = new MLFeatureEnabledSetting(clusterService, settings); - mlFeatureEnabledSetting.addListener(mlTaskManager); mlIndicesHandler = new MLIndicesHandler(clusterService, client, mlFeatureEnabledSetting); @@ -716,6 +715,7 @@ public Collection createComponents( this.mlStats = new MLStats(stats); mlTaskManager = new MLTaskManager(client, sdkClient, threadPool, mlIndicesHandler); + mlFeatureEnabledSetting.addListener(mlTaskManager); modelHelper = new ModelHelper(mlEngine); mlInputDatasetHandler = new MLInputDatasetHandler(client);