diff --git a/opensearch-search-quality-evaluation-framework/src/main/java/org/opensearch/eval/App.java b/opensearch-search-quality-evaluation-framework/src/main/java/org/opensearch/eval/App.java
index 7e148e5..f21a499 100644
--- a/opensearch-search-quality-evaluation-framework/src/main/java/org/opensearch/eval/App.java
+++ b/opensearch-search-quality-evaluation-framework/src/main/java/org/opensearch/eval/App.java
@@ -102,7 +102,7 @@ public static void main(String[] args) throws Exception {
                 final JsonObject jsonObject = jsonElement.getAsJsonObject();
                 final String samplerType = jsonObject.get("sampler").getAsString();
 
-                if("all".equalsIgnoreCase(samplerType)) {
+                if(AllQueriesQuerySampler.NAME.equalsIgnoreCase(samplerType)) {
 
                     final AllQueriesQuerySamplerParameters parameters = gson.fromJson(jsonString, AllQueriesQuerySamplerParameters.class);
 
@@ -111,7 +111,7 @@ public static void main(String[] args) throws Exception {
 
                     System.out.println("Query set created: " + querySetId);
 
-                } else if("pptss".equalsIgnoreCase(samplerType)) {
+                } else if(ProbabilityProportionalToSizeQuerySampler.NAME.equalsIgnoreCase(samplerType)) {
 
                     final ProbabilityProportionalToSizeParameters parameters = gson.fromJson(jsonString, ProbabilityProportionalToSizeParameters.class);