Fixed indexing bug for weights when more than 1 pol