diff --git a/module/base/button.py b/module/base/button.py
index a690888..d0bf122 100644
--- a/module/base/button.py
+++ b/module/base/button.py
@@ -231,9 +231,12 @@ class ButtonWrapper(Resource):
 
 
 class ClickButton:
-    def __init__(self, button, name='CLICK_BUTTON'):
-        self.area = button
-        self.button = button
+    def __init__(self, area, button=None, name='CLICK_BUTTON'):
+        self.area = area
+        if button is None:
+            self.button = area
+        else:
+            self.button = button
         self.name = name
 
     def __str__(self):