Weights registration bug solved