diff --git a/skywalking-agent/pom.xml b/skywalking-agent/pom.xml
index ef88cffa09ebeb96bbe72229ff040c7b43d7a11f..8a90f4cacca59ae470f6a3e9eba3d6be54d83e4c 100644
--- a/skywalking-agent/pom.xml
+++ b/skywalking-agent/pom.xml
@@ -16,6 +16,7 @@
 
     <properties>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+        <premain.class>com.ai.cloud.skywalking.agent.SkywalkingAgent</premain.class>
     </properties>
 
     <dependencies>
@@ -45,10 +46,7 @@
                                 </transformer>
                                 <transformer implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer">
                                     <manifestEntries>
-                                        <Premain-Class>com.ai.cloud.skywalking.agent.SkywalkingAgent</Premain-Class>
-                                        <!--<Main-Class>${app.main.class}</Main-Class>-->
-                                        <!--<X-Compile-Source-JDK>${maven.compile.source}</X-Compile-Source-JDK>-->
-                                        <!--<X-Compile-Target-JDK>${maven.compile.target}</X-Compile-Target-JDK>-->
+                                        <Premain-Class>${premain.class}</Premain-Class>
                                     </manifestEntries>
                                 </transformer>
                             </transformers>