Minor bug fix again weight should be changed (I know this is a bad commit, but...)