diff --git a/ballerina/build.gradle b/ballerina/build.gradle
index 3422c12..9d982bb 100644
--- a/ballerina/build.gradle
+++ b/ballerina/build.gradle
@@ -113,7 +113,7 @@ task startActiveMQServer() {
            if (!stdOut.toString().contains("activemq-test")) {
                println "Starting ActiveMQ server."
                exec {
-                   commandLine 'sh', '-c', "docker-compose -f tests/resources/docker-compose.yaml up -d"
+                   commandLine 'sh', '-c', "docker compose -f tests/resources/docker-compose.yaml up -d"
                    standardOutput = stdOut
                }
                println stdOut.toString()
@@ -136,7 +136,7 @@ task stopActiveMQServer() {
            if (stdOut.toString().contains("activemq-test")) {
                println "Stopping ActiveMQ server."
                exec {
-                   commandLine 'sh', '-c', "docker-compose -f tests/resources/docker-compose.yaml rm -svf"
+                   commandLine 'sh', '-c', "docker compose -f tests/resources/docker-compose.yaml rm -svf"
                    standardOutput = stdOut
                }
                println stdOut.toString()
diff --git a/ballerina/tests/resources/docker-compose.yaml b/ballerina/tests/resources/docker-compose.yaml
index 5eaf9cd..88cbfa5 100644
--- a/ballerina/tests/resources/docker-compose.yaml
+++ b/ballerina/tests/resources/docker-compose.yaml
@@ -1,5 +1,3 @@
-version: '2'
-
 services:
     activemq:
         image: 'apache/activemq-classic:6.0.0'
diff --git a/examples/ez-food/docker-compose.yaml b/examples/ez-food/docker-compose.yaml
index 9d180ea..7a8ce7d 100644
--- a/examples/ez-food/docker-compose.yaml
+++ b/examples/ez-food/docker-compose.yaml
@@ -1,5 +1,3 @@
-version: '2'
-
 services:
     order-svc:
         image: 'ballerina/ordersvc:1.0.0'