diff --git a/lodestone/__init__.py b/lodestone/__init__.py
index d87d112..a6273da 100644
--- a/lodestone/__init__.py
+++ b/lodestone/__init__.py
@@ -6,6 +6,7 @@
     from javascript import On as Event
     from lodestone.plugin import plugins
     from lodestone.logger import logger
+    from lodestone.gradios import ui
 except:
     from .bot import createBot, Bot
     from .utils import llm
@@ -13,4 +14,5 @@
     from .server import createServer, Server
     from javascript import On as Event
     from .plugin import plugins
-    from .logger import logger
\ No newline at end of file
+    from .logger import logger
+    from .gradios import ui
\ No newline at end of file
diff --git a/lodestone/__main__.py b/lodestone/__main__.py
index a7bbdc2..c451d43 100644
--- a/lodestone/__main__.py
+++ b/lodestone/__main__.py
@@ -6,7 +6,6 @@
 import time
 from rich.console import Console
 import lodestone
-from gradios import demo
 from dotenv import load_dotenv
 
 @click.command(context_settings={"ignore_unknown_options": True})
@@ -105,7 +104,7 @@ def check_python_command():
                 f.write(f"LODESTONE_USERNAME={username}\nLODESTONE_PASSWORD={password}")
                 print("Login details saved to .env file. You can change these at any time.")
         try:
-            demo.queue().launch(server_name=f"{host}", server_port=port, show_api=False, auth=(f'{os.environ["LODESTONE_USERNAME"]}', f'{os.environ["LODESTONE_PASSWORD"]}'), share=False, quiet=True)
+            lodestone.ui.queue().launch(server_name=f"{host}", server_port=port, show_api=False, auth=(f'{os.environ["LODESTONE_USERNAME"]}', f'{os.environ["LODESTONE_PASSWORD"]}'), share=False, quiet=True)
         except OSError:
             raise OSError(f"Port {port} is already in use!")
 
diff --git a/lodestone/gradios.py b/lodestone/gradios.py
index 697d1d0..25e403f 100644
--- a/lodestone/gradios.py
+++ b/lodestone/gradios.py
@@ -174,7 +174,7 @@ def build_schematic(files):
         print("not logged in!")
 
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as ui:
     with gr.Tab("Bot Settings"):
         # gr.Markdown(requests.get('https://raw.githubusercontent.com/the-lodestone-project/Lodestone/main/README.md').text)
         # gr.Image("https://github.com/the-lodestone-project/Lodestone/blob/main/assets/logo.png?raw=true", min_width=2000)
diff --git a/pyproject.toml b/pyproject.toml
index e471f72..025fbb1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "lodestone"
-version = "0.0.37"
+version = "0.0.38"
 description = "🤖 Create Minecraft bots with a powerful, stable, and high level Python API."
 authors = [
     { name = "Silke Pilon", email = "silkepilon2009@gmail.com" },