Make turn off actually work