Fix error in weights