diff --git a/src/Microsoft.ML.AutoML/SweepableEstimator/SweepablePipeline.cs b/src/Microsoft.ML.AutoML/SweepableEstimator/SweepablePipeline.cs
index 6f754d12b5..05dc88cd64 100644
--- a/src/Microsoft.ML.AutoML/SweepableEstimator/SweepablePipeline.cs
+++ b/src/Microsoft.ML.AutoML/SweepableEstimator/SweepablePipeline.cs
@@ -153,7 +153,7 @@ public string ToString(Parameter parameter)
             if (parameter.TryGetValue(AutoMLExperiment.PipelineSearchspaceName, out var pipelineParameter))
             {
                 var schema = pipelineParameter["_SCHEMA_"].AsType<string>();
-                var estimatorStrings = Entity.FromExpression(_currentSchema)
+                var estimatorStrings = Entity.FromExpression(schema)
                                    .ValueEntities()
                                    .Where(e => e is StringEntity se && se.Value != "Nil")
                                    .Select((se) =>