Improve wrap api