diff --git a/bigtop-manager-agent/src/test/java/org/apache/bigtop/manager/agent/grpc/interceptor/TaskInterceptorTest.java b/bigtop-manager-agent/src/test/java/org/apache/bigtop/manager/agent/grpc/interceptor/TaskInterceptorTest.java
index 3f7ec58c..f4cae16b 100644
--- a/bigtop-manager-agent/src/test/java/org/apache/bigtop/manager/agent/grpc/interceptor/TaskInterceptorTest.java
+++ b/bigtop-manager-agent/src/test/java/org/apache/bigtop/manager/agent/grpc/interceptor/TaskInterceptorTest.java
@@ -20,6 +20,7 @@
 
 import org.apache.bigtop.manager.common.utils.ProjectPathUtils;
 
+import org.apache.bigtop.manager.grpc.generated.TaskLogRequest;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -92,7 +93,7 @@ public void testTruncateLogFileFileDoesNotExist() throws IOException {
 
     @Test
     public void testIsTaskLogRequestTaskRequest() {
-        TaskLogRequest taskLogRequest = new TaskLogRequest();
+        TaskLogRequest taskLogRequest = TaskLogRequest.newBuilder().setTaskId(1L).build();
 
         Boolean result = taskInterceptor.isTaskRequest(taskLogRequest);
 
@@ -129,10 +130,4 @@ public Long getTaskId() {
             return 1L;
         }
     }
-
-    private static class TaskLogRequest {
-        public Long getTaskId() {
-            return 1L;
-        }
-    }
 }