¿Cómo omitir la extensión del archivo en el autocompletado de zsh?

Soy nuevo en zsh en Catalina. En el .zshrc, actualmente tengo

setopt CORRECT
setopt CORRECT_ALL

Cuando ejecuto un programa java desde la línea de comandos, sólo quiero:

% java ProgramName

Si escribo java Pro autocompletar añade el resto del nombre pero también añade .class . (O, si tengo el archivo fuente en el mismo directorio, añade el . después del nombre) y luego tengo que borrar .class (o pulsar el retroceso una vez) antes de poder ejecutar el programa.

¿Existe una forma de hacer que el autocompletado de zsh omita .class (y . ) pero sólo cuando el comando inicial es java ?

Preferiría no instalar algo como Oh My Zsh que podría alterar mis otras configuraciones de zsh.