fix bug'