1 files changed, 12 insertions, 16 deletions
diff --git a/chrome/test/data/extensions/samples/mappy/mappy_toolstrip.html b/chrome/test/data/extensions/samples/mappy/mappy_toolstrip.html
index 88d1d52..9573344 100755
--- a/chrome/test/data/extensions/samples/mappy/mappy_toolstrip.html
+++ b/chrome/test/data/extensions/samples/mappy/mappy_toolstrip.html
@@ -1,21 +1,6 @@
 <style>
-.chrome-toolstrip #map {
-  display: none;
-}
-
-.chrome-toolstrip #button {
-  display: block;
-}
-
-.chrome-mole #map {
-  display: block;
-}
-
-.chrome-mole #button {
-  display: none;
-}
-
 #map {
+  display: none;
   width: 512px;
   height: 512px;
 }
@@ -25,9 +10,20 @@
 <script>
 var maps_key = "ABQIAAAATfHumDbW3OmRByfquHd3SRTRERdeAiwZ9EeJWta3L_JZVS0bOBRQeZgr4K0xyVKzUdnnuFl8X9PX0w";
 
+chrome.toolstrip.onCollapsed.addListener(function() {
+  var map = document.getElementById("map");
+  map.src = "";
+  map.style.display = "none";
+  var button = document.getElementById("button");
+  button.style.display = "block";
+});
+
 function expand(url) {
+  var button = document.getElementById("button");
+  button.style.display = "none";
   var map = document.getElementById("map");
   map.src = url;
+  map.style.display = "block";
   chrome.toolstrip.expand({height:512}, function() {});
 }