Replace Shortcode in Divi / Extra

function extra_child_theme_setup() {
if ( class_exists('ET_Builder_Module')) {
get_template_part( 'cbm' );
$cbm = new WPC_ET_Builder_Module_Portfolio();
remove_shortcode( 'et_pb_portfolio' );
add_shortcode( 'et_pb_portfolio', array($cbm, '_shortcode_callback') );
}
}
add_action('wp', 'extra_child_theme_setup', 9999);