Admin toolbar – add new edit button for the custom posts

Hi.
I have added this code to add avia builder functionality to my custom posts (course_units):

add_filter(‘avf_builder_boxes’, ‘add_builder_to_posttype’);

function add_builder_to_posttype($metabox)
{
foreach($metabox as &$meta)
{
if($meta[‘id’] == ‘avia_builder’ || $meta[‘id’] == ‘layout’)
{
$meta[‘page’][] = ‘course_unit’; /*instead add the name of the custom post type here*/
}
}

return $metabox;
}

All is working great from the back end, except I can’t see edit post in the wordpress toolbar. What I mean is that I can’t edit my course post while I see it, because there is no button in the toolbar.

How can I enable it for my course_units posts?

I hope it all makes sense.

Thanks a lot

Darius