diff --git a/nix/base-system/init.lua b/nix/base-system/init.lua
index 778709f..26c8db1 100644
--- a/nix/base-system/init.lua
+++ b/nix/base-system/init.lua
@@ -503,18 +503,19 @@ local harpoon = require('harpoon')
 
 harpoon:setup()
 
-local lean = require'lean'
+local lean = require 'lean'
 
-lean.setup{
-  lsp = {
-      on_attach = function(client, bufnr)
-        local opts = { buffer = bufnr }
-        vim.keymap.set({ "n", "v" }, "<Leader>ti", function () vim.cmd("LeanInfoviewToggle") end, opts)
-        vim.keymap.set({ "n", "v" }, "<Leader>sg", function () vim.cmd("LeanGoal") end, opts)
-        vim.keymap.set({ "n", "v" }, "<Leader>stg", function () vim.cmd("LeanTermGoal") end, opts)
-      end
-  },
-  mappings = true,
+lean.setup {
+    lsp = {
+        on_attach = function(client, bufnr)
+            local opts = { buffer = bufnr }
+            vim.keymap.set({ "n", "v" }, "<Leader>ti", function() vim.cmd("LeanInfoviewToggle") end, opts)
+            vim.keymap.set({ "n", "v" }, "<Leader>sg", function() vim.cmd("LeanGoal") end, opts)
+            vim.keymap.set({ "n", "v" }, "<Leader>stg", function() vim.cmd("LeanTermGoal") end, opts)
+            vim.api.nvim_buf_set_option(bufnr, 'omnifunc', 'v:lua.vim.lsp.omnifunc')
+        end
+    },
+    mappings = true,
 }
 
 -- telescope keymaps