def METHOD_NAME(self) -> Optional[Union[Token, ast.AST]]:
    """
    Returns a node (comment or AST node) if it is in the line immediately after
    the docstring, otherwise returns None.
    """
    dostring_end_lineno = self.docstring_end_lineno
    comment = self.comments.get(dostring_end_lineno + 1, None)
    if comment is not None:
        return comment
    function_node = self.function_node
    second_node = function_node.body[1] if len(function_node.body) > 1 else None
    if second_node and second_node.lineno == dostring_end_lineno + 1:
        return second_node
    return None