1 files changed, 15 insertions, 15 deletions
diff --git a/gr-digital/python/ofdm.py b/gr-digital/python/ofdm.py
index 9f57920efc..00f52e3730 100644
--- a/gr-digital/python/ofdm.py
+++ b/gr-digital/python/ofdm.py
@@ -21,8 +21,8 @@
 # 
 
 import math
-from gnuradio import gr
-import digital_swig
+from gnuradio import gr, fft
+import digital_swig as digital
 import ofdm_packet_utils
 from ofdm_receiver import ofdm_receiver
 import gnuradio.gr.gr_threading as _threading
@@ -97,16 +97,16 @@ class ofdm_mod(gr.hier_block2):
             constel = qam.qam_constellation(arity)
             rotated_const = map(lambda pt: pt * rot, constel.points())
         #print rotated_const
-        self._pkt_input = digital_swig.ofdm_mapper_bcv(rotated_const,
-                                                       msgq_limit,
-                                                       options.occupied_tones,
-                                                       options.fft_length)
+        self._pkt_input = digital.ofdm_mapper_bcv(rotated_const,
+                                                  msgq_limit,
+                                                  options.occupied_tones,
+                                                  options.fft_length)
         
-        self.preambles = digital_swig.ofdm_insert_preamble(self._fft_length,
-                                                           padded_preambles)
-        self.ifft = gr.fft_vcc(self._fft_length, False, win, True)
-        self.cp_adder = digital_swig.ofdm_cyclic_prefixer(self._fft_length,
-                                                          symbol_length)
+        self.preambles = digital.ofdm_insert_preamble(self._fft_length,
+                                                      padded_preambles)
+        self.ifft = fft.fft_vcc(self._fft_length, False, win, True)
+        self.cp_adder = digital.ofdm_cyclic_prefixer(self._fft_length,
+                                                     symbol_length)
         self.scale = gr.multiply_const_cc(1.0 / math.sqrt(self._fft_length))
         
         self.connect((self._pkt_input, 0), (self.preambles, 0))
@@ -240,10 +240,10 @@ class ofdm_demod(gr.hier_block2):
 
         phgain = 0.25
         frgain = phgain*phgain / 4.0
-        self.ofdm_demod = digital_swig.ofdm_frame_sink(rotated_const, range(arity),
-                                                       self._rcvd_pktq,
-                                                       self._occupied_tones,
-                                                       phgain, frgain)
+        self.ofdm_demod = digital.ofdm_frame_sink(rotated_const, range(arity),
+                                                  self._rcvd_pktq,
+                                                  self._occupied_tones,
+                                                  phgain, frgain)
 
         self.connect(self, self.ofdm_recv)
         self.connect((self.ofdm_recv, 0), (self.ofdm_demod, 0))