diff --git a/README.md b/README.md
index 1f73fde..5431b6b 100644
--- a/README.md
+++ b/README.md
@@ -2,10 +2,10 @@
 
 ## Shell integration
 
-```
+```shell
 function cheat() { 
   bat -l md -p ~/cheat/cheats/"$1" --color always --theme="Dracula" | tr -d '`'
 }
 # autocompletion for zsh
-compdef '_files -W "/home/panki/cheat"' cheat
+compdef '_files -W "/home/panki/cheat/cheats"' cheat
 ```