diff --git a/src/test/java/com/checkmarx/ast/ResultTest.java b/src/test/java/com/checkmarx/ast/ResultTest.java
index 1f281ff2..b782d97c 100644
--- a/src/test/java/com/checkmarx/ast/ResultTest.java
+++ b/src/test/java/com/checkmarx/ast/ResultTest.java
@@ -36,7 +36,7 @@ void testResultsJSON() throws Exception {
         List<Scan> scanList = wrapper.scanList("statuses=Completed");
         Assertions.assertTrue(scanList.size() > 0);
         String scanId = scanList.get(0).getId();
-        String results = wrapper.results(UUID.fromString(scanId), ReportFormat.json);
+        String results = wrapper.results(UUID.fromString(scanId), ReportFormat.json, "java-wrapper");
         Assertions.assertTrue(results.length() > 0);
     }
 
@@ -53,9 +53,13 @@ void testResultsSummaryJSON() throws Exception {
     void testResultsStructure() throws Exception {
         List<Scan> scanList = wrapper.scanList("statuses=Completed");
         Assertions.assertTrue(scanList.size() > 0);
-        String scanId = scanList.get(0).getId();
-        Results results = wrapper.results(UUID.fromString(scanId));
-        Assertions.assertEquals(results.getTotalCount(), results.getResults().size());
+        for (Scan scan : scanList) {
+            Results results = wrapper.results(UUID.fromString(scan.getId()));
+            if (results != null && results.getResults() != null) {
+                Assertions.assertEquals(results.getTotalCount(), results.getResults().size());
+                break;
+            }
+        }
     }
 
     @Test()