diff --git a/lua/tiny-debugger/mode.lua b/lua/tiny-debugger/mode.lua
index 86f772d..1bf5d2b 100644
--- a/lua/tiny-debugger/mode.lua
+++ b/lua/tiny-debugger/mode.lua
@@ -1,6 +1,7 @@
 local config = require("tiny-debugger.config")
 local help = require("tiny-debugger.help")
 local ui = require("tiny-debugger.ui")
+local virtual_text = require("tiny-debugger.virtual_text")
 
 local M = {}
 
@@ -107,6 +108,7 @@ function M.enter()
   set_keymaps(buf)
   set_cursor_highlight()
   ui.open()
+  virtual_text.enable()
 
   if config.opts.help_on_enter then
     help.open()
@@ -122,6 +124,7 @@ function M.exit()
   restore_cursor_highlight()
   help.close()
   ui.close()
+  virtual_text.disable()
 end
 
 function M.toggle()
diff --git a/lua/tiny-debugger/virtual_text.lua b/lua/tiny-debugger/virtual_text.lua
new file mode 100644
index 0000000..3f1235c
--- /dev/null
+++ b/lua/tiny-debugger/virtual_text.lua
@@ -0,0 +1,24 @@
+local config = require("tiny-debugger.config")
+
+local M = {}
+
+local initialized = false
+
+function M.setup()
+  if initialized then return end
+  initialized = true
+  require("nvim-dap-virtual-text").setup(config.opts.virtual_text)
+end
+
+function M.enable()
+  M.setup()
+  require("nvim-dap-virtual-text").enable()
+end
+
+function M.disable()
+  if initialized then
+    require("nvim-dap-virtual-text").disable()
+  end
+end
+
+return M