diff --git a/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportGetTrainedModelsStatsAction.java b/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportGetTrainedModelsStatsAction.java
index 1f0506864de4a..dc04d65103f5f 100644
--- a/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportGetTrainedModelsStatsAction.java
+++ b/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportGetTrainedModelsStatsAction.java
@@ -214,10 +214,7 @@ protected void doExecuteForked(
                     responseBuilder.getExpandedModelIdsWithAliases(),
                     request.isAllowNoResources(),
                     parentTaskId,
-                    l.map(modelSizeStatsByModelId -> {
-                        responseBuilder.setModelSizeStatsByModelId(modelSizeStatsByModelId);
-                        return null;
-                    }),
+                    l,
                     numberOfAllocations
                 )
             )
diff --git a/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/utils/InferenceProcessorInfoExtractor.java b/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/utils/InferenceProcessorInfoExtractor.java
index 32d2fe40ea201..5a2f044d1f7be 100644
--- a/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/utils/InferenceProcessorInfoExtractor.java
+++ b/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/utils/InferenceProcessorInfoExtractor.java
@@ -13,7 +13,6 @@
 import org.elasticsearch.ingest.ConfigurationUtils;
 import org.elasticsearch.ingest.IngestMetadata;
 import org.elasticsearch.ingest.Pipeline;
-import org.elasticsearch.logging.LogManager;
 import org.elasticsearch.transport.Transports;
 
 import java.util.HashMap;
@@ -77,11 +76,7 @@ public static int countInferenceProcessors(ClusterState state) {
      */
     @SuppressWarnings("unchecked")
     public static Map<String, Set<String>> pipelineIdsByResource(ClusterState state, Set<String> ids) {
-        try {
-            assert Transports.assertNotTransportThread("non-trivial nested loops over cluster state structures");
-        } catch (AssertionError e) {
-            LogManager.getLogger(InferenceProcessorInfoExtractor.class).fatal("should not be here", e);
-        }
+        assert Transports.assertNotTransportThread("non-trivial nested loops over cluster state structures");
         Map<String, Set<String>> pipelineIdsByModelIds = new HashMap<>();
         Metadata metadata = state.metadata();
         if (metadata == null) {