diff --git a/lua/cp/config.lua b/lua/cp/config.lua
index 708969f..1a52a5c 100644
--- a/lua/cp/config.lua
+++ b/lua/cp/config.lua
@@ -139,8 +139,6 @@ function M.setup(user_config)
                 return false
               end
 
-              -- Allow any language and extension configurations
-
               return true
             end,
             'contest configuration',
diff --git a/lua/cp/init.lua b/lua/cp/init.lua
index 3c403fc..b24180c 100644
--- a/lua/cp/init.lua
+++ b/lua/cp/init.lua
@@ -140,7 +140,7 @@ local function setup_problem(contest_id, problem_id, language)
     config.hooks.setup_code(ctx)
   end
 
-  cache.set_file_state(ctx.source_file, state.platform, contest_id, problem_id, language)
+  cache.set_file_state(vim.fn.expand('%:p'), state.platform, contest_id, problem_id, language)
 
   logger.log(('switched to problem %s'):format(ctx.problem_name))
 end