fixup! RFNoC: More graph traversal fixes

2 files changed, 4 insertions, 4 deletions

diff --git a/host/lib/rfnoc/block_ctrl_base.cpp b/host/lib/rfnoc/block_ctrl_base.cpp
index 4137da9e3..ed8068b5b 100644
--- a/host/lib/rfnoc/block_ctrl_base.cpp
+++ b/host/lib/rfnoc/block_ctrl_base.cpp
@@ -125,8 +125,8 @@ block_ctrl_base::block_ctrl_base(const make_args_t& make_args)
     /*** Init I/O port definitions ******************************************/
     _init_port_defs("in", _block_def->get_input_ports());
     _init_port_defs("out", _block_def->get_output_ports());
-    set_num_input_ports(_block_def->get_input_ports().size());
-    set_num_output_ports(_block_def->get_output_ports().size());
+    _num_input_ports = _block_def->get_input_ports().size();
+    _num_output_ports = _block_def->get_output_ports().size();
     // FIXME this warning always fails until the input buffer code above is fixed
     // if (_tree->list(_root_path / "ports/in").size() != n_valid_input_buffers) {
     //    UHD_LOGGER_WARNING(unique_id()) <<
diff --git a/host/lib/rfnoc/source_block_ctrl_base.cpp b/host/lib/rfnoc/source_block_ctrl_base.cpp
index c3c6faf1f..6c41fae14 100644
--- a/host/lib/rfnoc/source_block_ctrl_base.cpp
+++ b/host/lib/rfnoc/source_block_ctrl_base.cpp
@@ -31,10 +31,10 @@ void source_block_ctrl_base::issue_stream_cmd(
     for (const node_ctrl_base::node_map_pair_t upstream_node : _upstream_nodes) {
         // FIXME:  Need proper mapping from input port to output port
         // The code below assumes the input port and output port are the same
-        // if the number of upstream and downstream connections are the same.
+        // if the number of upstream and downstream ports are the same.
         // The stream command is limited to only that port to prevent issuing
         // it on the wrong block and port.
-        if (get_num_input_ports() == get_num_output_ports()
+        if (_num_input_ports == _num_output_ports
             and upstream_node.first != chan) {
             continue;
         }