Heater Control Fix

Thanks!