fixing bug in code