Bug fixed

fixing bug