Fix weights after #17785

ArthurZ changed pull request status to merged