Add wrapping for a test