Remove unnecessary np.array()