3 files changed, 9 insertions, 4 deletions
diff --git a/gr-wavelet/python/CMakeLists.txt b/gr-wavelet/python/CMakeLists.txt
index 7f4d3a7af2..7f7fd99ecf 100644
--- a/gr-wavelet/python/CMakeLists.txt
+++ b/gr-wavelet/python/CMakeLists.txt
@@ -33,8 +33,12 @@ GR_PYTHON_INSTALL(
 if(ENABLE_TESTING)
 
 list(APPEND GR_TEST_PYTHON_DIRS
+    ${CMAKE_BINARY_DIR}/gr-analog/python
+    ${CMAKE_BINARY_DIR}/gr-analog/swig
     ${CMAKE_BINARY_DIR}/gr-wavelet/python
     ${CMAKE_BINARY_DIR}/gr-wavelet/swig
+    ${CMAKE_BINARY_DIR}/gr-blocks/python
+    ${CMAKE_BINARY_DIR}/gr-blocks/swig
 )
 list(APPEND GR_TEST_TARGET_DEPS gnuradio-wavelet)
 
diff --git a/gr-wavelet/python/__init__.py b/gr-wavelet/python/__init__.py
index 77e2ac7c26..cf86408e95 100644
--- a/gr-wavelet/python/__init__.py
+++ b/gr-wavelet/python/__init__.py
@@ -20,8 +20,7 @@
 #
 
 '''
-This is the gr-wavelet package. This package provides GNU Radio
-processing blocks for wavelet transforms.
+Processing blocks for wavelet transforms.
 '''
 
 from wavelet_swig import *
diff --git a/gr-wavelet/python/qa_classify.py b/gr-wavelet/python/qa_classify.py
index 5701bce8c1..c2972c92f2 100755
--- a/gr-wavelet/python/qa_classify.py
+++ b/gr-wavelet/python/qa_classify.py
@@ -26,6 +26,8 @@ import copy
 #import pygsl.wavelet as wavelet # FIXME: pygsl not checked for in config
 import math
 import wavelet_swig
+import analog_swig as analog
+import blocks_swig as blocks
 
 def sqr(x):
     return x*x
@@ -66,7 +68,7 @@ class test_classify(gr_unittest.TestCase):
         trg_data = src_data * 0.5
         src = gr.vector_source_f(src_data)
         dst = gr.vector_sink_f()
-        rail = gr.rail_ff(-0.5, 0.5)
+        rail = analog.rail_ff(-0.5, 0.5)
         self.tb.connect(src, rail)
         self.tb.connect(rail, dst)
         self.tb.run()
@@ -87,7 +89,7 @@ class test_classify(gr_unittest.TestCase):
         trg_data = copy.deepcopy(src_data)
 
         src = gr.vector_source_f(src_data, False, len(src_data))
-        st = gr.stretch_ff(-1.0/5.0, len(src_data))
+        st = blocks.stretch_ff(-1.0/5.0, len(src_data))
         dst = gr.vector_sink_f(len(src_data))
         self.tb.connect(src, st)
         self.tb.connect(st, dst)