diff --git a/riskified-sdk/src/main/java/com/riskified/models/Response.java b/riskified-sdk/src/main/java/com/riskified/models/Response.java
index 8233bd0..b2a09d2 100644
--- a/riskified-sdk/src/main/java/com/riskified/models/Response.java
+++ b/riskified-sdk/src/main/java/com/riskified/models/Response.java
@@ -44,7 +44,7 @@ public void setDecision(String decision) {
         this.decision = decision;
     }
 
-    public int getReceived() {
+    public Integer getReceived() {
         return received;
     }