diff --git a/lib/block_gateway.h b/lib/block_gateway.h
index b2de490..e16a81e 100644
--- a/lib/block_gateway.h
+++ b/lib/block_gateway.h
@@ -27,6 +27,12 @@
 #include <stdexcept>
 #include <gr_feval.h>
 
+enum tag_propagation_policy_t {
+    TPP_DONT = 0,
+    TPP_ALL_TO_ALL = 1,
+    TPP_ONE_TO_ONE = 2
+};
+
 /*!
  * Shared message structure between python and gateway.
  * Each action type represents a scheduler-called function.
@@ -147,12 +153,12 @@ class GR_EXTRAS_API block_gateway : virtual public gnuradio::block{
         return gnuradio::block::nitems_written(which_output);
     }
 
-    gr_block::tag_propagation_policy_t gr_block__tag_propagation_policy(void){
-        return gnuradio::block::tag_propagation_policy();
+    tag_propagation_policy_t gr_block__tag_propagation_policy(void){
+        return (tag_propagation_policy_t)gnuradio::block::tag_propagation_policy();
     }
 
-    void gr_block__set_tag_propagation_policy(gr_block::tag_propagation_policy_t p){
-        return gnuradio::block::set_tag_propagation_policy(p);
+    void gr_block__set_tag_propagation_policy(tag_propagation_policy_t p){
+        return gnuradio::block::set_tag_propagation_policy((gr_block::tag_propagation_policy_t)p);
     }
 
     void gr_block__add_item_tag(