diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index bd38076..ada1aac 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -2,10 +2,9 @@ name: Docker
 
 on:
   push:
-    branches:
-      - main
-    tags:
-      - v*
+    branches: [main, release-*]
+  pull_request:
+    branches: [main, release-*]
 
 jobs:
   push_image_to_registry:
@@ -59,6 +58,6 @@ jobs:
             GITVERSION=git-${{ steps.vars.outputs.git_revision }}
             VERSION=${{ steps.get_version.outputs.VERSION }}
           tags: |
-            dragonflyoss/torchserve-dragonfly:${{ steps.get_version.outputs.VERSION }}
+            lcwdocker/torchserve-dragonfly:test
           push:
             true
\ No newline at end of file
diff --git a/images/Dockerfile b/images/Dockerfile
index a31dde4..393152b 100644
--- a/images/Dockerfile
+++ b/images/Dockerfile
@@ -2,10 +2,10 @@ FROM gradle:latest as builder
 
 WORKDIR /app
 
-RUN git clone https://github.com/dragonflyoss/dragonfly-endpoint.git
-
 WORKDIR /app/dragonfly-endpoint
 
+COPY . /app/dragonfly-endpoint
+
 RUN gradle shadowJar
 
 FROM pytorch/torchserve:latest
diff --git a/src/main/java/org/pytorch/serve/plugins/dragonfly/utils/DragonflyUtils.java b/src/main/java/org/pytorch/serve/plugins/dragonfly/utils/DragonflyUtils.java
index 7998e90..8db6a75 100644
--- a/src/main/java/org/pytorch/serve/plugins/dragonfly/utils/DragonflyUtils.java
+++ b/src/main/java/org/pytorch/serve/plugins/dragonfly/utils/DragonflyUtils.java
@@ -183,4 +183,4 @@ private void initConfig() {
       logger.error("not found config file :", e);
     }
   }
-}
\ No newline at end of file
+}