diff --git a/rill-flow-task-template/src/test/groovy/com/weibo/rill/flow/task/template/dao/provider/TaskTemplateProviderTest.groovy b/rill-flow-task-template/src/test/groovy/com/weibo/rill/flow/task/template/dao/provider/TaskTemplateProviderTest.groovy
index 47a7e21e5..cb9500a24 100644
--- a/rill-flow-task-template/src/test/groovy/com/weibo/rill/flow/task/template/dao/provider/TaskTemplateProviderTest.groovy
+++ b/rill-flow-task-template/src/test/groovy/com/weibo/rill/flow/task/template/dao/provider/TaskTemplateProviderTest.groovy
@@ -33,11 +33,12 @@ class TaskTemplateProviderTest extends Specification {
         taskTemplateDO.setCategory("function")
         taskTemplateDO.setId(1L)
         taskTemplateDO.setType(0)
+        taskTemplateDO.setTaskYaml("{}")
         when:
         String sql = provider.update(taskTemplateDO)
         then:
         sql == "UPDATE task_template\n" +
-                "SET `type` = #{type}, `category` = #{category}\n" +
+                "SET `type` = #{type}, `category` = #{category}, `task_yaml` = #{taskYaml}\n" +
                 "WHERE (id = #{id})"
     }