diff --git a/ballerina/tests/tests.bal b/ballerina/tests/tests.bal
index 3fce618..d4cbb3a 100644
--- a/ballerina/tests/tests.bal
+++ b/ballerina/tests/tests.bal
@@ -54,7 +54,6 @@ isolated function testSendImageVariationRequest() returns error? {
     
     test:assertTrue(response is ImagesResponse, msg = "Response is not of type ImagesResponse");
 
-    return;
 }
 
 @test:Config {
@@ -79,8 +78,6 @@ isolated function testSendImageEditRequest() returns error? {
     var response = openAIImages->/images/edits.post(request);
    
     test:assertTrue(response is ImagesResponse, msg = "Response is not of type ImagesResponse");
-    
-    return;
 
 }
 
@@ -100,5 +97,4 @@ isolated function testSendImageGenerateRequest() returns error? {
 
     test:assertTrue(response is ImagesResponse, msg = "Response is not of type ImagesResponse");
     
-    return;
 }