update to integrate with updated wrapper