diff --git a/salty/base/neovim/init.sls b/salty/base/neovim/init.sls
index 15fce06..19ac055 100644
--- a/salty/base/neovim/init.sls
+++ b/salty/base/neovim/init.sls
@@ -16,7 +16,7 @@ neovim-install:
 python-neovim:
   pkg.installed:
     - pkgs:
-      - {{ grains.python2 }}
-      - {{ grains.python3 }}
+      - {{ neovim.python2 }}
+      - {{ neovim.python3 }}
 {% endif %}