Fix annotation button highlighting