diff --git a/lib-ce/emqx_modules/src/emqx_mod_trace_api.erl b/lib-ce/emqx_modules/src/emqx_mod_trace_api.erl
index 99fb97796..1daab1520 100644
--- a/lib-ce/emqx_modules/src/emqx_mod_trace_api.erl
+++ b/lib-ce/emqx_modules/src/emqx_mod_trace_api.erl
@@ -88,8 +88,11 @@ update_trace(Path, Params) ->
 download_zip_log(Path, Params) ->
     case emqx_trace_api:download_zip_log(Path, Params) of
         {ok, File} -> minirest:return_file(File);
-        {error, _Reason} = Err ->  return(Err)
+        {error, Reason} ->  return({error, 'NOT_FOUND', Reason})
     end.
 
 stream_log_file(Path, Params) ->
-    return(emqx_trace_api:stream_log_file(Path, Params)).
+    case emqx_trace_api:stream_log_file(Path, Params) of
+        {ok, File} -> return({ok, File});
+        {error, Reason} -> return({error, 'NOT_FOUND', Reason})
+    end.