il block contains methods fixed, resolves #3000

3 files changed, 3 insertions, 3 deletions

diff --git a/python/highlevelil.py b/python/highlevelil.py
index cb0a0980..176b2994 100644
--- a/python/highlevelil.py
+++ b/python/highlevelil.py
@@ -2556,7 +2556,7 @@ class HighLevelILBasicBlock(basicblock.BasicBlock):
 		return hash((self.start, self.end, self.il_function))
 
 	def __contains__(self, instruction):
-		if type(instruction) != HighLevelILInstruction or instruction.il_basic_block != self:
+		if not isinstance(instruction, HighLevelILInstruction) or instruction.il_basic_block != self:
 			return False
 		if self.start <= instruction.instr_index <= self.end:
 			return True
diff --git a/python/lowlevelil.py b/python/lowlevelil.py
index 8c246b73..22ea1e64 100644
--- a/python/lowlevelil.py
+++ b/python/lowlevelil.py
@@ -4716,7 +4716,7 @@ class LowLevelILBasicBlock(basicblock.BasicBlock):
 		return hash((self.start, self.end, self._il_function))
 
 	def __contains__(self, instruction):
-		if type(instruction) != LowLevelILInstruction or instruction.il_basic_block != self:
+		if not isinstance(instruction, LowLevelILInstruction) or instruction.il_basic_block != self:
 			return False
 		return True
 
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index fa0f6c4e..26610733 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -3229,7 +3229,7 @@ class MediumLevelILBasicBlock(basicblock.BasicBlock):
 		return hash((self.start, self.end, self._il_function))
 
 	def __contains__(self, instruction):
-		if type(instruction) != MediumLevelILInstruction or instruction.il_basic_block != self:
+		if not isinstance(instruction, MediumLevelILInstruction) or instruction.il_basic_block != self:
 			return False
 		if self.start <= instruction.instr_index <= self.end:
 			return True