fix import typo in cogroup.py