fix missing npz_fn bug