diff --git a/lua/persistence/init.lua b/lua/persistence/init.lua
index 6b1702a..4bfc985 100644
--- a/lua/persistence/init.lua
+++ b/lua/persistence/init.lua
@@ -46,6 +46,7 @@ function M.start()
 end
 
 function M.stop()
+  vim.g.using_persistance = nil
   vim.cmd([[
     autocmd! Persistence
     augroup! Persistence
@@ -57,12 +58,14 @@ function M.save()
   vim.o.sessionoptions = table.concat(Config.options.options, ",")
   vim.cmd("mks! " .. e(M.get_current()))
   vim.o.sessionoptions = tmp
+  vim.g.using_persistance = e(M.get_current())
 end
 
 function M.load(opt)
   opt = opt or {}
   local sfile = opt.last and M.get_last() or M.get_current()
   if sfile and vim.fn.filereadable(sfile) ~= 0 then
+    vim.g.using_persistance = e(sfile)
     vim.cmd("source " .. e(sfile))
   end
 end