diff --git a/DrissionPage/common.py b/DrissionPage/common.py
index 4f50e68..75e90c3 100644
--- a/DrissionPage/common.py
+++ b/DrissionPage/common.py
@@ -13,7 +13,7 @@ from selenium.webdriver.remote.webelement import WebElement
 
 
 class DrissionElement(object):
-    def __init__(self, ele):
+    def __init__(self, ele: Union[Element, WebElement]):
         self._inner_ele = ele
 
     @property