Fix some flow graph and disassembly API issues

1 files changed, 8 insertions, 1 deletions

diff --git a/python/flowgraph.py b/python/flowgraph.py
index 103eed69..4d2bd3ba 100644
--- a/python/flowgraph.py
+++ b/python/flowgraph.py
@@ -78,7 +78,7 @@ class FlowGraphNode(object):
 
 	@property
 	def basic_block(self):
-		"""Basic block associated with this part of the flow graph (read-only)"""
+		"""Basic block associated with this part of the flow graph"""
 		block = core.BNGetFlowGraphBasicBlock(self.handle)
 		if not block:
 			return None
@@ -100,6 +100,13 @@ class FlowGraphNode(object):
 			block = basicblock.BasicBlock(block, view)
 		return block
 
+	@basic_block.setter
+	def basic_block(self, block):
+		if block is None:
+			core.BNSetFlowGraphBasicBlock(self.handle, None)
+		else:
+			core.BNSetFlowGraphBasicBlock(self.handle, block.handle)
+
 	@property
 	def x(self):
 		"""Flow graph block X (read-only)"""