Требуется редактировать элементы дерева в модальном диалоге, но при нажатии на Enter диалог закрывается :(

Пробовал лечить при помощи PreTranslateMessage, но получилось что-то не то:
редактируемый текст после нажатия изменяется на первоначальный. Как это можно исправить?