DiffFactorizer: modify some more code