diff --git a/plugins/utils/undotree.nix b/plugins/utils/undotree.nix
index 11b36054..5c74cc4b 100644
--- a/plugins/utils/undotree.nix
+++ b/plugins/utils/undotree.nix
@@ -9,6 +9,12 @@ in
     plugins.undotree = {
       enable = mkEnableOption "Enable undotree";
 
+      package = mkOption {
+        type = types.package;
+        default = pkgs.vimPlugins.undotree;
+        description = "Plugin to use for undotree";
+      };
+
       windowLayout = mkOption {
         type = types.nullOr types.int;
         default = null;
@@ -108,7 +114,7 @@ in
   };
 
   config = mkIf cfg.enable {
-    extraPlugins = [ pkgs.vimPlugins.undotree ];
+    extraPlugins = [ cfg.package ];
 
     globals = {
       undotree_WindowLayout = mkIf (cfg.windowLayout != null) cfg.windowLayout;