Update unit.rb

bug fix