Fixing bug in update notification