Fixing broken collection update