Adding testPrepareBenchmarkStage