diff --git a/src/networks/model-net/dragonfly-custom.C b/src/networks/model-net/dragonfly-custom.C
index e66e4c0fe88bb6c4c9da371fece4bd0f876b0347..d2cbd688e305d4e209edc1f2ac531ce297878f5a 100644
--- a/src/networks/model-net/dragonfly-custom.C
+++ b/src/networks/model-net/dragonfly-custom.C
@@ -366,7 +366,6 @@ struct router_state
    int* cur_hist_num;
    
    char output_buf[4096];
-   char output_buf2[4096];
 
    struct dfly_router_sample * rsamples;
    
@@ -2294,24 +2293,25 @@ void dragonfly_custom_router_final(router_state * s,
                 s->busy_time[d]);
         }
     }
-
         map< int, vector<bLink> >  &curMap = interGroupLinks[s->router_id];
         map< int, vector<bLink> >::iterator it = curMap.begin();
         for(; it != curMap.end(); it++)
         {
-            int grp_id = it->first;
-            int dest_rtr_id = interGroupLinks[s->router_id][grp_id][0].dest;
-            /* TODO: Works only for 1-D dragonfly right now. Make it functional
+            /* TODO: Works only for single global connections right now. Make it functional
              * for a 2-D dragonfly. */
-            int offset = interGroupLinks[s->router_id][grp_id][0].offset;
-            written += sprintf(s->output_buf + written, "\n%d %s %d %s %s %llu %lf", 
-                s->router_id,
-                "R",
-                dest_rtr_id,
-                "R",
-                "G",
-                s->link_traffic[offset],
-                s->busy_time[offset]);
+            for(int l = 0; l < it->second.size(); l++) {
+                int dest_rtr_id = it->second[l].dest;
+                int offset = it->second[l].offset;
+                assert(offset >= 0 && offset < p->num_global_channels);
+                written += sprintf(s->output_buf + written, "\n%d %s %d %s %s %llu %lf", 
+                    s->router_id,
+                    "R",
+                    dest_rtr_id,
+                    "R",
+                    "G",
+                    s->link_traffic[offset],
+                    s->busy_time[offset]);
+            }
         }
     sprintf(s->output_buf + written, "\n");
     lp_io_write(lp->gid, (char*)"dragonfly-link-stats", written, s->output_buf);
@@ -3277,30 +3277,23 @@ router_packet_receive( router_state * s,
 
   output_chan = 0;
   if(output_port < s->params->intra_grp_radix) {
-   if(DF_DALLY == 0)
+   if(DF_DALLY == 1)
    {
     if(cur_chunk->msg.my_g_hop == 1) {
-     if(cur_chunk->msg.my_l_hop < 1) {
-        cur_chunk->msg.my_l_hop = 1;
-      }
+        output_chan = 1;
     } else if (cur_chunk->msg.my_g_hop == 2) {
-      if(cur_chunk->msg.my_l_hop < 4) {
-        cur_chunk->msg.my_l_hop = 4;
-      }
+        output_chan = 3;
     }
    }
   else {
-    /*if(cur_chunk->msg.my_g_hop == 1) {
-        if(cur_chunk->msg.my_l_hop < 1){
-            cur_chunk->msg.my_l_hop = 1;
-        }
+      /* TODO: Recheck VC count after things are in order for a 2-D dragonfly. */
+    if(cur_chunk->msg.my_g_hop == 1) {
+        output_chan = 2;
     }
-    else*/ if (cur_chunk->msg.my_g_hop == 2) {
-      if(cur_chunk->msg.my_l_hop < 2)
-        cur_chunk->msg.my_l_hop = 2;
+    else if (cur_chunk->msg.my_g_hop == 2) {
+        output_chan = 6;
     }
   }
-    output_chan = cur_chunk->msg.my_l_hop;
     max_vc_size = s->params->local_vc_size;
     cur_chunk->msg.my_l_hop++;
   } else if(output_port < (s->params->intra_grp_radix +