diff --git a/includes/admin/assets/css/um-admin-global.css b/includes/admin/assets/css/um-admin-global.css
index a87874fe..c33b5c1f 100644
--- a/includes/admin/assets/css/um-admin-global.css
+++ b/includes/admin/assets/css/um-admin-global.css
@@ -203,8 +203,18 @@ a.um-delete{ color: #a00; }
 	margin-bottom: 2px !important;
 }
 
+/*
+	- Restrict access
+*/
+
 .um_hidden_notice,
-.um_block_settings_hide {
+.um_block_settings .um_block_settings_hide{
+	display: none;
+}
+.um_block_settings .components-select-control .components-select-control__input[multiple]{
+	height: auto;
+}
+.um_block_settings .components-select-control .components-select-control__input[multiple] + .components-input-control__suffix{
 	display: none;
 }
 
diff --git a/includes/admin/assets/js/um-admin-blocks-shortcode.js b/includes/admin/assets/js/um-admin-blocks-shortcode.js
index bf6981c2..88038cf3 100644
--- a/includes/admin/assets/js/um-admin-blocks-shortcode.js
+++ b/includes/admin/assets/js/um-admin-blocks-shortcode.js
@@ -365,7 +365,7 @@ wp.blocks.registerBlockType( 'um-block/um-account', {
 				wp.i18n.__( 'Account', 'ultimate-member' )
 			),
 			wp.element.createElement(
-				wp.editor.InspectorControls,
+				wp.blockEditor.InspectorControls,
 				{},
 				wp.element.createElement(
 					wp.components.PanelBody,
diff --git a/includes/admin/assets/js/um-admin-blocks.js b/includes/admin/assets/js/um-admin-blocks.js
index c9dc2aa0..9322e816 100644
--- a/includes/admin/assets/js/um-admin-blocks.js
+++ b/includes/admin/assets/js/um-admin-blocks.js
@@ -60,12 +60,13 @@ var um_block_restriction = wp.compose.createHigherOrderComponent( function( Bloc
 			{},
 			wp.element.createElement( BlockEdit, props ),
 			wp.element.createElement(
-				wp.editor.InspectorControls,
+				wp.blockEditor.InspectorControls,
 				{},
 				wp.element.createElement(
 					wp.components.PanelBody,
 					{
-						title: wp.i18n.__( 'UM access Controls', 'ultimate-member' )
+						title: wp.i18n.__( 'UM access Controls', 'ultimate-member' ),
+						className: 'um_block_settings'
 					},
 					wp.element.createElement(
 						wp.components.ToggleControl,