diff --git a/lua/outline/ui.lua b/lua/outline/highlight.lua
similarity index 100%
rename from lua/outline/ui.lua
rename to lua/outline/highlight.lua
diff --git a/lua/outline/init.lua b/lua/outline/init.lua
index 87cdae9..0219080 100644
--- a/lua/outline/init.lua
+++ b/lua/outline/init.lua
@@ -1,7 +1,7 @@
 local Sidebar = require('outline.sidebar')
 local cfg = require('outline.config')
+local highlight = require('outline.highlight')
 local providers = require('outline.providers.init')
-local ui = require('outline.ui')
 local utils = require('outline.utils.init')
 
 local M = {
@@ -242,7 +242,7 @@ function M.setup(opts)
   }
 
   cfg.setup(opts)
-  ui.setup_highlights()
+  highlight.setup_highlights()
 
   setup_global_autocmd()
   setup_commands()
diff --git a/lua/outline/writer.lua b/lua/outline/writer.lua
index 1c38b43..3d80763 100644
--- a/lua/outline/writer.lua
+++ b/lua/outline/writer.lua
@@ -1,9 +1,9 @@
 local cfg = require('outline.config')
 local folding = require('outline.folding')
+local highlight = require('outline.highlight')
 local parser = require('outline.parser')
 local symbols = require('outline.symbols')
 local t_utils = require('outline.utils.table')
-local ui = require('outline.ui')
 
 local strlen = vim.fn.strlen
 
@@ -47,14 +47,14 @@ function M.add_hover_highlights(bufnr, nodes)
   end
 
   -- clear old highlight
-  ui.clear_hover_highlight(bufnr)
+  highlight.clear_hover_highlight(bufnr)
   for _, node in ipairs(nodes) do
     if not node.hovered then
       goto continue
     end
 
     if node.prefix_length then
-      ui.add_hover_highlight(bufnr, node.line_in_outline - 1, node.prefix_length)
+      highlight.add_hover_highlight(bufnr, node.line_in_outline - 1, node.prefix_length)
     end
     ::continue::
   end