Fix call to BNDefineAutoSymbolAndVariableOrFunction

1 files changed, 8 insertions, 2 deletions

diff --git a/python/binaryview.py b/python/binaryview.py
index a48880dd..aef72a6a 100644
--- a/python/binaryview.py
+++ b/python/binaryview.py
@@ -3287,7 +3287,7 @@ class BinaryView(object):
 		.. warning:: If multiple symbols for the same address are defined, only the most recent symbol will ever be used.
 
 		:param Symbol sym: the symbol to define
-		:param SymbolType sym_type: Type of symbol being defined
+		:param SymbolType sym_type: Type of symbol being defined (can be None)
 		:param Platform plat: (optional) platform
 		:rtype: None
 		"""
@@ -3295,7 +3295,13 @@ class BinaryView(object):
 			plat = self.platform
 		elif not isinstance(plat, binaryninja.platform.Platform):
 			raise AttributeError("Provided platform is not of type `binaryninja.platform.Platform`")
-		core.BNDefineAutoSymbolAndVariableOrFunction(self.handle, plat.handle, sym.handle, sym_type.handle)
+
+		if isinstance(sym_type, binaryninja.SymbolType):
+			sym_type = sym_type.handle
+		elif sym_type is not None:
+			raise AttributeError("Provided sym_type is not of type `binaryninja.SymbolType`")
+
+		core.BNDefineAutoSymbolAndVariableOrFunction(self.handle, plat.handle, sym.handle, sym_type)
 
 	def undefine_auto_symbol(self, sym):
 		"""