Warmup fix

@jpmccabe