Minor UI fix (keep buttons the same throughout)