diff --git a/src/block/divider/edit.js b/src/block/divider/edit.js
index 093a6fbb38..e8b8864248 100644
--- a/src/block/divider/edit.js
+++ b/src/block/divider/edit.js
@@ -137,5 +137,5 @@ export default compose(
 	withBlockWrapperIsHovered,
 	withQueryLoopContext,
 	withBlockAttributeContext,
-	withBlockStyleContext( blockStyles ),
+	withBlockStyleContext( dividerStyles ),
 )( Edit )
diff --git a/src/block/icon-button/edit.js b/src/block/icon-button/edit.js
index 1b21ab3ed8..9974b34f6d 100644
--- a/src/block/icon-button/edit.js
+++ b/src/block/icon-button/edit.js
@@ -121,5 +121,5 @@ export default compose(
 	withBlockWrapper,
 	withQueryLoopContext,
 	withBlockAttributeContext,
-	withBlockStyleContext( blockStyles ),
+	withBlockStyleContext( iconButtonStyles ),
 )( Edit )