diff --git a/examples/base.py b/examples/base.py
index a5f1f19..1ae5aa3 100644
--- a/examples/base.py
+++ b/examples/base.py
@@ -45,7 +45,7 @@
 dataset = pybi.duckdb.from_pandas({"df": pd.DataFrame(data)})
 
 
-@pybi.page()
+@ui.page()
 def index():
     table = dataset["df"]
 
diff --git a/src/pybi/__init__.py b/src/pybi/__init__.py
index 1ef8d9b..a508090 100644
--- a/src/pybi/__init__.py
+++ b/src/pybi/__init__.py
@@ -2,9 +2,7 @@
     "__version__",
     "data_view",
     "query",
-    "page",
     "label",
-    "server",
     "duckdb",
     "column",
     "row",
@@ -17,9 +15,8 @@
 ]
 
 from .version import __version__
-from instaui.ui import page, server, column, row, label, grid
+from instaui.ui import column, row, label, grid
 from .link_sql import duckdb
-from instaui.arco import *  # noqa: F403
 from .components.radio import radio
 from .components.select import select
 from .components.table import table