BugFix: store the weigths to the right location