Fix regression caused by numpy array removal