diff --git a/src/finn/qnn-data/templates/driver/driver_base.py b/src/finn/qnn-data/templates/driver/driver_base.py
index 2096760580b4f33ba1ab09564ebba1601c4dc23c..5f6f00da13de551a3e3cd234ad28458ca3cc6287 100644
--- a/src/finn/qnn-data/templates/driver/driver_base.py
+++ b/src/finn/qnn-data/templates/driver/driver_base.py
@@ -189,14 +189,10 @@ class FINNExampleOverlay(Overlay):
             layer_ind = int(w_filename.split("_")[1])
             rt_weight_dict[(sdp_ind, layer_ind)] = layer_w
         for sdp_ind, layer_ind in rt_weight_dict.keys():
-            cand_if_name = "StreamingDataflowPartition_%d/s_axilite_%d" % (
-                sdp_ind,
-                layer_ind,
-            )
+            cand_if_name = "StreamingDataflowPartition_%d" % sdp_ind
             if cand_if_name in self.ip_dict.keys():
                 layer_mmio = getattr(
-                    getattr(self, "StreamingDataflowPartition_%d" % sdp_ind),
-                    "s_axilite_%d" % layer_ind,
+                    self, "StreamingDataflowPartition_%d" % sdp_ind
                 ).mmio
                 layer_w = rt_weight_dict[(sdp_ind, layer_ind)]
                 layer_mmio.write_mm(0, layer_w.tobytes())