diff --git a/httprunner/__init__.py b/httprunner/__init__.py
index 3376f842..8e62f016 100644
--- a/httprunner/__init__.py
+++ b/httprunner/__init__.py
@@ -1 +1 @@
-__version__ = '0.8.1a'
\ No newline at end of file
+__version__ = '0.8.1b'
\ No newline at end of file
diff --git a/httprunner/cli.py b/httprunner/cli.py
index b2c488a4..f90b8ab6 100644
--- a/httprunner/cli.py
+++ b/httprunner/cli.py
@@ -4,8 +4,9 @@ import os
 import sys
 from collections import OrderedDict
 
-import PyUnitReport
-from PyUnitReport import __version__ as pyu_version
+from pyunitreport import __version__ as pyu_version
+from pyunitreport import HTMLTestRunner
+
 from httprunner import __version__ as ate_version
 from httprunner import exception
 from httprunner.task import TaskSuite
@@ -77,7 +78,7 @@ def main_ate():
             "report_name": report_name,
             "failfast": args.failfast
         }
-        result = PyUnitReport.HTMLTestRunner(**kwargs).run(task_suite)
+        result = HTMLTestRunner(**kwargs).run(task_suite)
         results[testset_path] = OrderedDict({
             "total": result.testsRun,
             "successes": len(result.successes),