Fixes pytest branch for new cpy_wrapper API