Remove Divi Builder Button for WP User Role Author

To make things easier for the client hand-off, I removed the Divi builder button from the back-end, using the following code:

/* Remove Divi Builder button for WP Role Author */
function my_custom_admin_head() {
  if ( current_user_can( 'author' )) :
  ?><style>.et_pb_toggle_builder_wrapper { display: none; }</style>
  <?php endif;
}
add_action( 'admin_head', 'my_custom_admin_head' );