fix set lookup_fn bug