In that case it was me just reading the diff outside of the full context. Code looks good.