- if ( wp_update_term($cat_ID, 'link_category', $_POST) )
- wp_redirect('edit-link-categories.php?message=3');
+ $location = 'edit-link-categories.php';
+ if ( $referer = wp_get_original_referer() ) {
+ if ( false !== strpos($referer, 'edit-link-categories.php') )
+ $location = $referer;
+ }
+
+ $update = wp_update_term($cat_ID, 'link_category', $_POST);
+
+ if ( $update && !is_wp_error($update) )
+ $location = add_query_arg('message', 3, $location);