WPML ATE - inherit custom permalink

function pm_ate_inherit_permalink($post_id, $fields) {
    global $permalink_manager_uris, $sitepress;

    if(class_exists('Permalink_Manager_URI_Functions_Post')) {
        $original_post_id = $sitepress->get_original_element_id($post_id);

        if(!empty($permalink_manager_uris[$original_post_id])) {
            $permalink_manager_uris[$post_id] = $permalink_manager_uris[$original_post_id];;
            update_option('permalink-manager-uris', $permalink_manager_uris);
        }
    }
}
add_action('icl_pro_translation_completed', 'pm_ate_inherit_permalink', 150, 2);