fix warmup logic