diff --git a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs
index 8c5613014e..2b7feb1dcf 100644
--- a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs
+++ b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs
@@ -197,10 +197,10 @@ public static T TrainerKindApiValue<T>(TrainerKinds trainerKind)
 
         public static bool IsTrainerOfKind(Type type, TrainerKinds trainerKind)
         {
-            if (type == typeof(Trainers.BinaryLogisticRegressor))
+            if (type == typeof(Trainers.LogisticRegressionBinaryClassifier))
                 return trainerKind == TrainerKinds.SignatureBinaryClassifierTrainer;
 
-            if (type == typeof(Trainers.LogisticRegressor))
+            if (type == typeof(Trainers.LogisticRegressionClassifier))
                 return trainerKind == TrainerKinds.SignatureMultiClassClassifierTrainer;
 
             if (trainerKind != TrainerKinds.SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds.SignatureMultiOutputRegressorTrainer)