fixing bug

fixing bug