Very minor optimization to DIIS