Forgot to update comment