Fix weights after#17785

ArthurZ changed pull request status to merged