Fix tests for wrappers