diff --git a/tests/args_files_enabled_spec.lua b/tests/args_files_enabled_spec.lua
index 7a29389..e598ad9 100644
--- a/tests/args_files_enabled_spec.lua
+++ b/tests/args_files_enabled_spec.lua
@@ -90,7 +90,7 @@ describe("The args files enabled config", function()
   end)
 
   it("doesn't autosave when args_allow_files_auto_save returns false", function()
-    M.clearSessionFilesAndBuffers()
+    TL.clearSessionFilesAndBuffers()
 
     vim.cmd("e " .. TL.other_file)
     assert.equals(1, vim.fn.bufexists(TL.other_file))
@@ -111,7 +111,7 @@ describe("The args files enabled config", function()
   end)
 
   it("does autosave a session when args_allow_files_auto_save returns true", function()
-    M.clearSessionFilesAndBuffers()
+    TL.clearSessionFilesAndBuffers()
 
     vim.cmd("e " .. TL.other_file)
     assert.equals(1, vim.fn.bufexists(TL.other_file))
diff --git a/tests/cmds_custom_dir_spec.lua b/tests/cmds_custom_dir_spec.lua
index 8af6818..2dfdf59 100644
--- a/tests/cmds_custom_dir_spec.lua
+++ b/tests/cmds_custom_dir_spec.lua
@@ -8,7 +8,7 @@ describe("The default config", function()
   }
 
   local custom_sessions_dir = vim.fn.getcwd() .. "/tests/custom_sessions/"
-  local cwd_session_name = M.escapeSessionName(vim.fn.getcwd())
+  local cwd_session_name = TL.escapeSessionName(vim.fn.getcwd())
   local cwd_session_path = custom_sessions_dir .. cwd_session_name .. ".vim"
   local named_session_path = custom_sessions_dir .. TL.named_session_name .. ".vim"