[benchmarking] BUGFIX reverted missing import statements