1 files changed, 6 insertions, 0 deletions
diff --git a/plugins/pychrysa/analysis/module.c b/plugins/pychrysa/analysis/module.c
index 05c2e2e..0dfb5e0 100644
--- a/plugins/pychrysa/analysis/module.c
+++ b/plugins/pychrysa/analysis/module.c
@@ -26,7 +26,10 @@
 
 
 #include "binary.h"
+#include "block.h"
+#include "routine.h"
 #include "binaries/module.h"
+#include "blocks/module.h"
 
 
 
@@ -61,7 +64,10 @@ bool add_analysis_module_to_python_module(PyObject *super)
     result = (ret == 0);
 
     result &= register_python_loaded_binary(module);
+    result &= register_python_binary_routine(module);
+    result &= register_python_instructions_block(module);
     result &= add_binaries_module_to_python_module(module);
+    result &= add_blocks_module_to_python_module(module);
 
     return result;