fix toggle return when button is pressed