Remove redundant weight reset