diff --git a/httprunner/step.py b/httprunner/step.py
index 8c42aea5..75cc43dc 100644
--- a/httprunner/step.py
+++ b/httprunner/step.py
@@ -14,6 +14,8 @@ from httprunner.step_sql_request import (
     StepSqlRequestExtraction,
 )
 
+from httprunner.step_thrift_request import RunThriftRequest,StepThriftRequestValidation,StepThriftRequestExtraction
+
 
 class Step(object):
     def __init__(
@@ -26,6 +28,9 @@ class Step(object):
             RunSqlRequest,
             StepSqlRequestValidation,
             StepSqlRequestExtraction,
+            RunThriftRequest,
+            StepThriftRequestValidation,
+            StepThriftRequestExtraction
         ],
     ):
         self.__step = step
diff --git a/httprunner/thrift/data_convertor.py b/httprunner/thrift/data_convertor.py
index 45514926..b25af390 100644
--- a/httprunner/thrift/data_convertor.py
+++ b/httprunner/thrift/data_convertor.py
@@ -356,8 +356,8 @@ class ThriftJSONEncoder(json.JSONEncoder):
 
     def encode(self, o):
         """Return a JSON string representation of a Python data structure.
-        JSONEncoder().encode({"foo": ["bar", "baz"]})
-       '{"foo": ["bar", "baz"]}'
+         JSONEncoder().encode({"foo": ["bar", "baz"]})
+        '{"foo": ["bar", "baz"]}'
 
         """
         # This is for extremely simple cases and benchmarks.