Fix weights issue