diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 736dbdb..db2fa4e 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -13,7 +13,7 @@ jobs:
           fetch-depth: 0
 
       - name: golangci-lint
-        uses: golangci/golangci-lint-action@v2
+        uses: golangci/golangci-lint-action@v3
         with:
           version: latest
 
@@ -28,12 +28,14 @@ jobs:
         fetch-depth: 0
 
     - name: Install Go
-      uses: actions/setup-go@v3
+      uses: actions/setup-go@v4
       with:
-        go-version: "1.20"
+        go-version: "1.21"
 
     - name: Test
-      run: go test ./...
+      uses: gomicro/forge-action@main
+      with:
+        step: test
 
   deploy:
     name: Deploy
diff --git a/forge.yaml b/forge.yaml
index 8c9dafc..f87181e 100644
--- a/forge.yaml
+++ b/forge.yaml
@@ -1,5 +1,8 @@
 project:
   name: train
+envs:
+  DOCKER_PASSWORD: $DOCKER_PASSWORD
+  DOCKER_USERNAME: $DOCKER_USERNAME
 steps:
   build:
     help: Build the project
@@ -14,7 +17,18 @@ steps:
       -o {{.Project}} .
   clean:
     help: Clean up all generated files
-    cmd: go clean
+    cmds:
+      - go clean
+      - rm -f coverage.txt
+  coverage:
+    help: generate coverage
+    cmd: docker run -v $PWD:/src gomicro/gocover
+  deploy:
+    help: deploy the artifacts
+    cmds:
+      - echo "Logging into Docker Hub"
+      - echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin
+      - "VERSION={{.ShortSha}} goreleaser release"
   fmt:
     help: Run gofmt
     cmd: go fmt ./...
@@ -34,7 +48,7 @@ steps:
       GOOS: '{{.Os}}'
     cmd: >
       go install -ldflags
-      "-X 'github.com/gomicro/train/cmd.Version=dev-$(git rev-parse --short HEAD)'
+      "-X 'github.com/gomicro/train/cmd.Version=dev-{{.ShortSha}}'
       -X 'github.com/gomicro/train/cmd.clientID=$TRAIN_CLIENT_ID'
       -X 'github.com/gomicro/train/cmd.clientSecret=$TRAIN_CLIENT_SECRET'"
   test: