Only show buttons based on the current state