refactor warmup. We no need no ifs