diff --git a/modules/python/init.zsh b/modules/python/init.zsh
index d61488ed9a..a8e00b39b8 100644
--- a/modules/python/init.zsh
+++ b/modules/python/init.zsh
@@ -160,7 +160,7 @@ if (( $#commands[(i)pip(|[23])] )); then
         || ! -s "$cache_file" ]]; then
     # pip is slow; cache its output. And also support 'pip2', 'pip3' variants
     $pip_command completion --zsh \
-      | sed -e "s|\(compctl -K [-_[:alnum:]]*\) pip.*|\1 pip pip2 pip3|" >! "$cache_file" 2> /dev/null
+      | sed -e "s/\(compctl -K [-_[:alnum:]]*\) pip.*/\1 pip(|[23](|.[0-9]))/" >! "$cache_file" 2> /dev/null
   fi
 
   source "$cache_file"