1 files changed, 4 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