Fix using 2 buttons