diff --git a/chi/__init__.py b/chi/__init__.py
index 0bacc10..098beda 100644
--- a/chi/__init__.py
+++ b/chi/__init__.py
@@ -10,7 +10,7 @@
     nova,
     zun,
 )
-from .context import get, params, reset, session, set, use_site, use_device_auth
+from .context import get, params, reset, session, set, use_device_auth, use_site
 
 __all__ = [
     "get",