fix waitgroup logic

@wvanbergen @manygrams

There were some issues with shutting down the `ConsumerGroup`. This seems to fix them.