diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index c7da453..cc2d3fb 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -3,7 +3,7 @@ name: Upload Python Package
 on:
   push:
     branches:
-      - test
+      - main
 
 permissions:
   contents: read
diff --git a/pyproject.toml b/pyproject.toml
index ead404f..0a949d5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 
 [project]
 name = "streamlit-auth"
-version = "0.1.7"
+version = "0.1.7.1"
 description = "A Python package for creating oauth protected Streamlit apps"
 authors = [
     {name = "Hugo Zhu", email = "hugozhu@gmail.com"}
diff --git a/src/streamlit_auth/__init__.py b/src/streamlit_auth/__init__.py
index b7229c7..e5e5dbc 100644
--- a/src/streamlit_auth/__init__.py
+++ b/src/streamlit_auth/__init__.py
@@ -14,8 +14,7 @@
 
 def new_session_id():
     return uuid.uuid4().hex
-
-@st.cache_data
+    
 def get_manager():
     return stx.CookieManager()