fixing bug.....