diff options
Diffstat (limited to 'python/undoaction.py')
| -rw-r--r-- | python/undoaction.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/undoaction.py b/python/undoaction.py index 850d37fd..9f742e00 100644 --- a/python/undoaction.py +++ b/python/undoaction.py @@ -24,6 +24,7 @@ import ctypes # Binary Ninja components import _binaryninjacore as core +from enums import ActionType import startup import log @@ -40,7 +41,7 @@ class UndoAction(object): raise TypeError("undo action type not registered") action_type = self.__class__.action_type if isinstance(action_type, str): - self._cb.type = core.BNActionType[action_type] + self._cb.type = ActionType[action_type] else: self._cb.type = action_type self._cb.context = 0 |