Update reset function

Fix reset bug.