1 files changed, 7 insertions, 8 deletions
diff --git a/gr-trellis/src/lib/trellis.i b/gr-trellis/src/lib/trellis.i
index 3a364802d..60bb831b1 100644
--- a/gr-trellis/src/lib/trellis.i
+++ b/gr-trellis/src/lib/trellis.i
@@ -20,21 +20,27 @@
  * Boston, MA 02110-1301, USA.
  */
 
+#define DIGITAL_API
+
 %include "gnuradio.i"			// the common stuff
 
 //load generated python docstrings
 %include "trellis_swig_doc.i"
 
 %{
+#include "digital/constellation.h"
+#include "digital/metric_type.h"
 #include "fsm.h"
 #include "interleaver.h"
 #include "trellis_permutation.h"
 #include "trellis_siso_f.h"
 #include "trellis_siso_combined_f.h"
 #include "trellis_constellation_metrics_cf.h"
-#include "digital/constellation.h"
 %}
 
+%include "digital/constellation.h"
+%include "digital/metric_type.h"
+%include "constellation.i"
 
 // ----------------------------------------------------------------
 
@@ -49,10 +55,3 @@
 %include "trellis_constellation_metrics_cf.i"
 
 %include "trellis_generated.i"
-
-%import "digital/metric_type.h"
-%import "constellation.i"
-
- //%pythoncode %{
- //  from gnuradio.gr import TRELLIS_EUCLIDEAN, TRELLIS_HARD_SYMBOL, TRELLIS_HARD_BIT
- //  %}