fixing bug
fixng bug