Good catch, will update this PR