Examples: updt api

1 files changed, 5 insertions, 4 deletions

diff --git a/example/expression/constant_propagation.py b/example/expression/constant_propagation.py
index d9c5fe65..0798c404 100644
--- a/example/expression/constant_propagation.py
+++ b/example/expression/constant_propagation.py
@@ -10,7 +10,8 @@ from miasm2.arch.x86.disasm import dis_x86_32 as dis_engine
 from miasm2.analysis.machine import Machine
 from miasm2.analysis.binary import Container
 from miasm2.analysis.cst_propag import propagate_cst_expr
-from miasm2.analysis.data_flow import dead_simp
+from miasm2.analysis.data_flow import dead_simp, \
+    merge_blocks, remove_empty_assignblks
 from miasm2.expression.simplifications import expr_simp
 
 
@@ -33,6 +34,7 @@ addr = int(args.address, 0)
 
 asmcfg = mdis.dis_multiblock(addr)
 ircfg = ir_arch.new_ircfg_from_asmcfg(asmcfg)
+entry_points = set([mdis.loc_db.get_offset_location(addr)])
 
 init_infos = ir_arch.arch.regs.regs_init
 cst_propag_link = propagate_cst_expr(ir_arch, ircfg, addr, init_infos)
@@ -43,9 +45,8 @@ if args.simplify:
     while modified:
         modified = False
         modified |= dead_simp(ir_arch, ircfg)
-        modified |= ircfg.remove_empty_assignblks()
-        modified |= ircfg.remove_jmp_blocks()
-        modified |= ircfg.merge_blocks()
+        modified |= remove_empty_assignblks(ircfg)
+        modified |= merge_blocks(ircfg, entry_points)
 
 
 open("%s.propag.dot" % args.filename, 'w').write(ircfg.dot())