Hack to fix benchmark