Unable to retain leaf icon after node editing in JTree

[Sudhakar Krishnamurthy]

Guys:
Here's my problem, I am trying to edit a JTree node name with customized icons. But as soon as i finish editing the customized icon is being replaced with the default leaf icon

I am using extending my CellEditor class from DefaultTreeCellEditor and providing it my own renderer extended from DefaultTreeCellRenderer. In my cell editor class i've overridden the "determineOffset" function.
Any help is appreciated....
TIA