update wrapper for testing