improve the reset stage a little