diff --git a/tabpy/tabpy_server/management/state.py b/tabpy/tabpy_server/management/state.py
index c19182b9..d5aeae35 100644
--- a/tabpy/tabpy_server/management/state.py
+++ b/tabpy/tabpy_server/management/state.py
@@ -342,10 +342,8 @@ def update_endpoint(
                 dependencies, endpoint_info.get("dependencies", []))
             # Adding is_public means that some existing functions do not have is_public set.
             # We need to check for this when updating and set to False by default
-            current_is_public = False
-            if hasattr(endpoint_info, "is_public"):
-                current_is_public = endpoint_info["is_public"]
-            is_public = self._check_and_set_is_public(is_public, current_is_public)
+            is_public = self._check_and_set_is_public(
+                is_public, getattr(endpoint_info, "is_public", False))
 
             self._check_target(target)
             if target and target not in endpoints: