Fix broken build because of beehive's api change