modify the ruby script

modify the ruby script