diff --git a/.github/workflows/selftest.yml b/.github/workflows/selftest.yml
index 9e49651..24906b8 100644
--- a/.github/workflows/selftest.yml
+++ b/.github/workflows/selftest.yml
@@ -7,17 +7,17 @@ jobs:
     name: Self-test
     runs-on: ubuntu-latest
     steps:
-      - name: Install gotestfmt
-        uses: haveyoudebuggedit/gotestfmt-action@main
       - name: Set up Go
         uses: actions/setup-go@v2
         with:
           go-version: 1.16
       - name: Checkout
         uses: actions/checkout@v2
+      - name: Install gotestfmt
+        uses: ./
       - name: Run gotestfmt
         working-directory: testdata
-        run: go test -v ./... 2>&1 | tee /tmp/gotest.log | gotestfmt
+        run: go test -json -v ./... 2>&1 | tee /tmp/gotest.log | gotestfmt
       - name: Upload test log
         uses: actions/upload-artifact@v2
         if: always()
diff --git a/README.md b/README.md
index 4a7e890..b874164 100644
--- a/README.md
+++ b/README.md
@@ -26,12 +26,12 @@ jobs:
 
       # Install gotestfmt on the VM running the action.
       - name: Set up gotestfmt
-        uses: haveyoudebuggedit/gotestfmt-action@v1
+        uses: haveyoudebuggedit/gotestfmt-action@v2
         with:
           # Optional: pass GITHUB_TOKEN to avoid rate limiting.
           token: ${{ secrets.GITHUB_TOKEN }}
           # Optional: pass the gotestfmt version you want to run. 
-          version: v1.0.0
+          version: v2.0.0
           # Optional: pass an organization name and repo to use a fork
           org: haveyoudebuggedit
           repo: gotestfmt
diff --git a/index.js b/index.js
index f3ad9a4..054e634 100644
--- a/index.js
+++ b/index.js
@@ -72,7 +72,7 @@ async function downloadGofmt(octokit, version, versionPrefix, org, repo, token)
 async function main() {
     try {
         // versionPrefix is the prefix of the version gotestfmt-action supports.
-        const versionPrefix = "v1."
+        const versionPrefix = "v2."
         const token = core.getInput('token');
         const version = core.getInput('version');
         const org = core.getInput("org")