diff --git a/bootstrap-wysiwyg.js b/bootstrap-wysiwyg.js
index 69f64a7..dc84b14 100644
--- a/bootstrap-wysiwyg.js
+++ b/bootstrap-wysiwyg.js
@@ -26,8 +26,9 @@
 			updateToolbar = function () {
 				if (options.activeToolbarClass) {
 					$(options.toolbarSelector).find(toolbarBtnSelector).each(function () {
-						var command = $(this).data(options.commandRole);
-						if (document.queryCommandState(command)) {
+						var command = $(this).data(options.commandRole),
+							format = command.match(/formatBlock (.+)/);
+						if (document.queryCommandState(command) || (format && (format[1] == document.queryCommandValue("formatBlock")))) {
 							$(this).addClass(options.activeToolbarClass);
 						} else {
 							$(this).removeClass(options.activeToolbarClass);