fix bug.....