diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 8ce60bb..4e13c10 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -19,15 +19,8 @@ jobs:
           go-version-file: 'go.mod'
           cache: true
 
-      # TODO: Uncomment this when we have fixed the ability to run the
-      # integration tests on the github actions runner.
-      # - name: Setup Ko
-      #   uses: ko-build/setup-ko@v0.9
-
-      # TODO: Uncomment this when we have fixed the ability to run the
-      # integration tests on the github actions runner.
-      # - name: Set up Docker
-      #   uses: docker/setup-docker-action@v4
+      - name: Setup Ko
+        uses: ko-build/setup-ko@v0.9
 
       - name: Install Task
         uses: arduino/setup-task@v2
@@ -38,7 +31,5 @@ jobs:
       - name: Test
         run: task test
 
-      # TODO: Uncomment this when we have fixed the ability to run the
-      # integration tests on the github actions runner.
-      # - name: Test
-      #   run: task test-integration
\ No newline at end of file
+      - name: Integration Test
+        run: task test-integration
diff --git a/Taskfile.yml b/Taskfile.yml
index 66d466e..8688de8 100644
--- a/Taskfile.yml
+++ b/Taskfile.yml
@@ -83,7 +83,7 @@ tasks:
   build-image:
     desc: Build the image locally with ko
     env:
-      KO_DOCKER_REPO: ghcr.io/stackloklabs/yardstick/server
+      KO_DOCKER_REPO: ghcr.io/stackloklabs/yardstick
       VERSION: dev-local
     cmds:
       - |
diff --git a/test/integration-test.sh b/test/integration-test.sh
index 8b6efdf..5d957f7 100755
--- a/test/integration-test.sh
+++ b/test/integration-test.sh
@@ -27,7 +27,7 @@ else
 fi
 
 # Get the image name from ko build output
-IMAGE_NAME="ghcr.io/stackloklabs/yardstick/server:latest"
+IMAGE_NAME="ghcr.io/stackloklabs/yardstick:latest"
 
 ###################################################################
 ################## START - STDIO TRANSPORT TESTING ################