diff --git a/tests/WP_Ultimo/SSO/Admin_Bar_Magic_Links_Test.php b/tests/WP_Ultimo/SSO/Admin_Bar_Magic_Links_Test.php index e249238b..9933a0bb 100644 --- a/tests/WP_Ultimo/SSO/Admin_Bar_Magic_Links_Test.php +++ b/tests/WP_Ultimo/SSO/Admin_Bar_Magic_Links_Test.php @@ -104,8 +104,12 @@ public function test_modify_my_sites_menu_uses_lazy_action_urls(): void { }; add_filter('wu_magic_link_url', $magic_link_filter); - $this->magic_links->modify_my_sites_menu($admin_bar); - remove_filter('wu_magic_link_url', $magic_link_filter); + + try { + $this->magic_links->modify_my_sites_menu($admin_bar); + } finally { + remove_filter('wu_magic_link_url', $magic_link_filter); + } $dashboard_node = $admin_bar->get_node('blog-' . $site_id . '-d'); $site_node = $admin_bar->get_node('blog-' . $site_id . '-c');