fix treatment overwrite bug