In reply to Martin Wolf.

Ah, my fault. The button is not showing for drafts, which makes sense. My bad, code is working fine!