diff --git a/hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/schema/Label.java b/hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/schema/Label.java
index da6996f3..5f5c9dd8 100644
--- a/hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/schema/Label.java
+++ b/hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/schema/Label.java
@@ -20,11 +20,11 @@ public interface Label extends Identifiable, LabelData {
   @GraphQLName(LABELED_ENTITIES_QUERY_NAME)
   LabeledEntityResultSet labeledEntities(
       @GraphQLNonNull @GraphQLName(ENTITY_TYPE_ARGUMENT_NAME) String entityType,
-      @GraphQLName(LimitArgument.ARGUMENT_NAME) int limit);
+      @GraphQLName(LimitArgument.ARGUMENT_NAME) Integer limit);
 
   @GraphQLField
   @GraphQLNonNull
   @GraphQLName(LABEL_APPLICATION_RULES_QUERY_NAME)
   LabelApplicationRuleResultSet labelApplicationRules(
-      @GraphQLName(LimitArgument.ARGUMENT_NAME) int limit);
+      @GraphQLName(LimitArgument.ARGUMENT_NAME) Integer limit);
 }
diff --git a/hypertrace-graphql-labels-schema-impl/src/main/java/org/hypertrace/graphql/label/dao/LabelResponseConverter.java b/hypertrace-graphql-labels-schema-impl/src/main/java/org/hypertrace/graphql/label/dao/LabelResponseConverter.java
index e0322fae..5b82749b 100644
--- a/hypertrace-graphql-labels-schema-impl/src/main/java/org/hypertrace/graphql/label/dao/LabelResponseConverter.java
+++ b/hypertrace-graphql-labels-schema-impl/src/main/java/org/hypertrace/graphql/label/dao/LabelResponseConverter.java
@@ -108,12 +108,12 @@ private static class DefaultLabel implements Label {
     LabelApplicationRuleResultSet labelApplicationRuleResultSet;
 
     @Override
-    public LabeledEntityResultSet labeledEntities(String entityType, int limit) {
+    public LabeledEntityResultSet labeledEntities(String entityType, Integer limit) {
       return labeledEntityResultSetMap.get(entityType);
     }
 
     @Override
-    public LabelApplicationRuleResultSet labelApplicationRules(int limit) {
+    public LabelApplicationRuleResultSet labelApplicationRules(Integer limit) {
       return labelApplicationRuleResultSet;
     }
   }