diff --git a/lua/persisted/init.lua b/lua/persisted/init.lua
index 14542df..6e61cd6 100644
--- a/lua/persisted/init.lua
+++ b/lua/persisted/init.lua
@@ -89,10 +89,6 @@ end
 function M.setup(opts)
   config.setup(opts)
 
-  if M.session_exists() then
-    vim.g.persisted_exists = true
-  end
-
   if
     config.options.autosave
     and (allow_dir() and not ignore_dir() and vim.g.persisting == nil)
diff --git a/lua/persisted/utils.lua b/lua/persisted/utils.lua
index d862685..2024927 100644
--- a/lua/persisted/utils.lua
+++ b/lua/persisted/utils.lua
@@ -86,6 +86,7 @@ function M.load_session(session, silent)
     return echoerr("Error loading the session! ", result)
   end
 
+  vim.g.persisted_exists = true
   vim.g.persisted_loaded_session = session
   vim.api.nvim_exec_autocmds("User", { pattern = "PersistedLoadPost", data = session })
 end