Fix reset button issue