Add testing wrapper