repair some bug :smiley: