Add wrapper test