diff --git a/chain_service/database/models/chain.py b/chain_service/database/models/chain.py
index 4be50ac..91c967c 100644
--- a/chain_service/database/models/chain.py
+++ b/chain_service/database/models/chain.py
@@ -5,7 +5,11 @@ from pydantic import BaseModel, Field
 from typing import Literal, Union, Annotated, Optional, List
 
 
-class WaitAction(BaseModel):
+class BaseAction(BaseModel):
+    action_type: Literal["wait", "comment"]
+
+
+class WaitAction(BaseAction):
     action_type: Literal["wait"]
     wait_for: int
 
@@ -13,17 +17,15 @@ class WaitAction(BaseModel):
         pass
 
 
-class TextCommentAction(BaseModel):
-    action_type: Literal["text_comment"]
+class CommentAction(BaseAction):
+    action_type: Literal["comment"]
     text: str
 
     class Config(BaseConfig):
         pass
 
 
-Action = Annotated[
-    Union[WaitAction, TextCommentAction], Field(description="action_type")
-]
+Action = Annotated[Union[WaitAction, CommentAction], Field(description="action_type")]
 
 
 class Chain(BaseMongoModel):