[FIR] Fix erasing existing data flow statements

This commit is contained in:
Dmitriy Novozhilov
2020-04-10 15:14:27 +03:00
parent 3c33ee9c68
commit 300c3c30fb
@@ -317,7 +317,7 @@ abstract class PersistentLogicSystem(context: ConeInferenceContext) : LogicSyste
if (shouldRemoveOriginalStatements) {
logicStatements -= originalVariable
}
logicStatements = logicStatements.put(newVariable, newStatements)
logicStatements = logicStatements.put(newVariable, logicStatements[newVariable]?.let { it + newStatements } ?: newStatements)
}
}