fixing bug again