Update to use new get_parent method