fixing bug with notification