fix shutoff when done