address feedback

EasyPost · Jul 23, 2024 · 527eaeb · 527eaeb
1 parent f846cd1
commit 527eaeb
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/main/java/com/easypost/service/ClaimService.java b/src/main/java/com/easypost/service/ClaimService.java
@@ -58,9 +58,18 @@ public Claim retrieve(final String id) throws EasyPostException {
      * @throws EasyPostException when the request fails.
      */
     public ClaimCollection all(final Map<String, Object> params) throws EasyPostException {
+        String type = (String) params.get("type");
+        String status = (String) params.get("status");
+        params.remove(type);
+        params.remove(status);
         String endpoint = "claims";
 
-        return Requestor.request(RequestMethod.GET, endpoint, params, ClaimCollection.class, client);
+        ClaimCollection claimCollection = 
+            Requestor.request(RequestMethod.GET, endpoint, params, ClaimCollection.class, client);
+        claimCollection.setType(type);
+        claimCollection.setType(status);
+
+        return claimCollection;
     }
 
     /**
@@ -72,7 +81,7 @@ public ClaimCollection all(final Map<String, Object> params) throws EasyPostExce
      */
     public Claim cancel(final String id) throws EasyPostException {
         String endpoint = String.format("claims/%s/cancel", id);
-        
+
         return Requestor.request(RequestMethod.POST, endpoint, null, Claim.class, client);
     }