diff --git a/lua/plugins/treesitter.lua b/lua/plugins/treesitter.lua
index 9432e7f..01ce348 100644
--- a/lua/plugins/treesitter.lua
+++ b/lua/plugins/treesitter.lua
@@ -3,9 +3,14 @@ return {
     'nvim-treesitter/nvim-treesitter',
     dependencies = {
       'nvim-treesitter/nvim-treesitter-textobjects', -- maybe try mini.ai instead?
+      'nvim-treesitter/nvim-treesitter-context',
     },
     build = ':TSUpdate',
     main = 'nvim-treesitter.configs', -- Sets main module to use for opts
+    config = function(_, opts)
+      require('nvim-treesitter.configs').setup(opts)
+      require('treesitter-context').setup(opts.context)
+    end,
     opts = {
       auto_install = false,
       highlight = { enable = true },
@@ -72,6 +77,9 @@ return {
           include_surrounding_whitespace = true,
         },
       },
+      context = {
+        enable = true,
+      },
     },
   },
 }