Do you need the np.array here?