Fix remove groups

This won't be found by `unittest` unless you put `__init__.py` everywhere.