class LogMode < TextMode
   register_keymap do |k|
     k.add :toggle_follow, "Toggle follow mode", 'f'
-    k.add :save_to_disk, "Save log to disk", 's'
   end
 
   def initialize
     end
   end
 
-  def save_to_disk
-    fn = BufferManager.ask_for_filename :filename, "Save log to file: "
-    save_to_file(fn) { |f| f.puts text } if fn
-  end
-
   def status
     super + " (follow: #@follow)"
   end
 
 
 class TextMode < ScrollMode
   attr_reader :text
+  register_keymap do |k|
+    k.add :save_to_disk, "Save to disk", 's'
+  end
 
   def initialize text=""
     @text = text.normalize_whitespace
     buffer.mark_dirty if buffer
     super()
   end
+  
+  def save_to_disk
+    fn = BufferManager.ask_for_filename :filename, "Save to file: "
+    save_to_file(fn) { |f| f.puts text } if fn
+  end
 
   def text= t
     @text = t