Setting weights of correct container