diff --git a/httprunner/step_thrift_request.py b/httprunner/step_thrift_request.py
index fe569032..7eef3e07 100644
--- a/httprunner/step_thrift_request.py
+++ b/httprunner/step_thrift_request.py
@@ -189,7 +189,7 @@ class RunThriftRequest(IStep):
 
         return self
 
-    def setup_hook(self, hook: Text, assign_var_name: Text = None) -> "RunTestCase":
+    def setup_hook(self, hook: Text, assign_var_name: Text = None) -> "RunThriftRequest":
         if assign_var_name:
             self.__step.setup_hooks.append({assign_var_name: hook})
         else: