[PATCH] Fix diff-pruning logic which was running prune too early.