diff --git a/mediation-remove-query-param/tests/utest_policy.bal b/mediation-remove-query-param/tests/utest_policy.bal
index 94dfa1e..5a2ee90 100644
--- a/mediation-remove-query-param/tests/utest_policy.bal
+++ b/mediation-remove-query-param/tests/utest_policy.bal
@@ -29,8 +29,8 @@ public function testRequestFlowSingleInstance() {
 @test:Config {}
 public function testRequestFlowMultipleInstance() {
     mediation:Context ctx = createContext("get", "/greet");
-    ctx.addQueryParam("arr", 1);
-    ctx.addQueryParam("arr", 2);
+    ctx.addQueryParam("arr", "1");
+    ctx.addQueryParam("arr", "2");
     ctx.addQueryParam("x", "y");
     http:Response|false|error|() result = removeQueryParam(ctx, new, "arr");
     assertResult(result, ctx.queryParams(), {x: ["y"]});