Add missing override for the on_remove_node method