La mia soluzione era molto simile, semplicemente disabilitava il pulsante che comuque rimaneva visibile... Se anche il tuo metodo funziona è ancora meglio, puoi scegliere...
Problema risolto

Ciao