Fix numpy.vstack bug