fixed typo in np.save