lib/progress.rb in progress-1.1.0 vs lib/progress.rb in progress-1.1.1

@@ -89,11 +89,10 @@
       if levels.last
         set(levels.last.current + Float(num) / den, &block)
       elsif block
         block.call
       end
-      self.note = nil
     end
 
     def set(value, &block)
       if levels.last
         ret = if block
@@ -101,10 +100,11 @@
         end
         if levels.last
           levels.last.current = Float(value)
         end
         print_message
+        self.note = nil
         ret
       elsif block
         block.call
       end
     end
@@ -213,10 +213,10 @@
 
         eta_string = eta(inner)
         message = "#{parts.reverse * ' > '}#{eta_string}"
         message_cl = "#{parts_cl.reverse * ' > '}#{eta_string}"
 
-        if note = levels.last.note
+        if note = levels.last && levels.last.note
           message << " - #{note}"
           message_cl << " - #{note}"
         end
 
         unless lines?