fixing bug.