Fix use of vector to store neighbour states