diff --git a/src/main/java/com/iflytop/gd/app/controller/SelfTestController.java b/src/main/java/com/iflytop/gd/app/controller/SelfTestController.java
index 41566fc..ddba9f4 100644
--- a/src/main/java/com/iflytop/gd/app/controller/SelfTestController.java
+++ b/src/main/java/com/iflytop/gd/app/controller/SelfTestController.java
@@ -30,10 +30,10 @@ public class SelfTestController {
         return Result.success(selfTestService.getSelfTestState());
     }
 
-    @Operation(summary = "自检完毕")
+    @Operation(summary = "是否自检完毕")
     @PostMapping("/finish")
-    public Result<?> selfTestFinish() throws Exception {
-        deviceStateService.setSelfTest(true);
+    public Result<?> selfTestFinish(Boolean mode) throws Exception {
+        deviceStateService.setSelfTest(mode);
         new Thread(() -> {
             try {
                 deviceCommandUtilService.capMotorMove(21);