fix bug..