remove unecessary imports in benchmark