Update wrapper for testing