diff --git a/library/think/app.php b/library/think/app.php
index f9f4e9e9..646e5831 100644
--- a/library/think/app.php
+++ b/library/think/app.php
@@ -115,7 +115,7 @@ class App
                 // 操作方法执行完成监听
                 APP_HOOK && Hook::listen('action_end', $data);
                 // 返回数据
-                if (defined('IN_UNIT_TEST')) {
+                if (IN_UNIT_TEST) {
                     return $data;
                 } else {
                     Response::returnData($data, Config::get('default_return_type'), Config::get('response_exit'));
@@ -132,7 +132,7 @@ class App
                 // 操作方法执行完成监听
                 APP_HOOK && Hook::listen('action_end', $data);
                 // 返回数据
-                if (defined('IN_UNIT_TEST')) {
+                if (IN_UNIT_TEST) {
                     return $data;
                 } else {
                     Response::returnData($data, Config::get('default_return_type'), Config::get('response_exit'));