make it work with the latest kafka