Bug fix, size of array