correct weights
correct weights