2 files changed, 5 insertions, 4 deletions
diff --git a/lib/oxidized/model/junos.rb b/lib/oxidized/model/junos.rb
index d1f4de6..72f5275 100644
--- a/lib/oxidized/model/junos.rb
+++ b/lib/oxidized/model/junos.rb
@@ -15,12 +15,12 @@ class JunOS < Oxidized::Model
   cmd 'show configuration'
 
   cmd 'show version' do |cfg|
-    chassis = model $1 if cfg.match /^Model: (\S+)/
-    comment cfg << chassis.to_s
+    @model = $1 if cfg.match /^Model: (\S+)/
+    comment cfg
   end
 
-  def model chassis
-    case chassis
+  def main
+    case @model
     when 'mx960'
       cmd('show chassis fabric reachability')  { |cfg| comment cfg }
     end
diff --git a/lib/oxidized/model/model.rb b/lib/oxidized/model/model.rb
index 00a520d..b768bfb 100644
--- a/lib/oxidized/model/model.rb
+++ b/lib/oxidized/model/model.rb
@@ -62,6 +62,7 @@ module Oxidized
         out = instance_exec out, &cmd_block if cmd_block
         data << out.to_s
       end
+      data << main.to_s if respond_to? :main
       data
     end