diff --git a/src/src/main.js b/src/src/main.js
index 72f2b4f..dac0de8 100644
--- a/src/src/main.js
+++ b/src/src/main.js
@@ -7,7 +7,7 @@ import CommonMixin from "@/mixins/commonMixin"
 import common from "@/utils/common"
 import tippy from "tippy.js"
 
-onUiLoaded(() => {
+onUiLoaded(async () => {
     const div = document.createElement('div')
     div.id    = 'physton-prompt-all-in-one'
     common.gradioApp().appendChild(div)