diff -r ba5f64fe0e00 -r d73b64672238 controls/CustomTree.py --- a/controls/CustomTree.py Wed Aug 16 12:30:31 2017 +0300 +++ b/controls/CustomTree.py Wed Aug 16 12:48:08 2017 +0300 @@ -35,6 +35,7 @@ self._type = (1 if image is not None else 0) self._ExtraImage = image + CT.GenericTreeItem.SetExtraImage = SetExtraImage _DefaultGetCurrentCheckedImage = CT.GenericTreeItem.GetCurrentCheckedImage @@ -44,6 +45,8 @@ if self._ExtraImage is not None: return self._ExtraImage return _DefaultGetCurrentCheckedImage(self) + + CT.GenericTreeItem.GetCurrentCheckedImage = GetCurrentCheckedImage