Edited TestBase for warmup