fix bug........