The only change here is this guard. Looks good.