Fix add_mark bug