diff --git a/lib/ComicBook.js b/lib/ComicBook.js
index 85dce82..b73a0b9 100755
--- a/lib/ComicBook.js
+++ b/lib/ComicBook.js
@@ -184,7 +184,7 @@ function ComicBook(id, srcs, opts) {
 			.append(
 				$(document.createElement("button"))
 					.text("layout")
-					.addClass(options.displayMode)
+					.addClass("cb-layout " + options.displayMode)
 					.click(function(){
 						ComicBook.prototype.toggleLayout();
 					})
@@ -334,10 +334,13 @@ function ComicBook(id, srcs, opts) {
 
 	ComicBook.prototype.toggleLayout = function() {
 		if (options.displayMode === "double") {
+			$("#cb-toolbar .cb-layout").removeClass("double");
 			options.displayMode = "single";
 		} else {
+			$("#cb-toolbar .cb-layout").removeClass("single");
 			options.displayMode = "double";
 		}
+		$("#cb-toolbar .cb-layout").addClass(options.displayMode);
 		ComicBook.prototype.draw();
 	};