Fix what we toggle when you click toggle