Non avevo notato....

Devi mettere il focus() prima del return.