Fix copy/paste for np.ones