diff --git a/src/finn/transformation/fpgadataflow/annotate_resources.py b/src/finn/transformation/fpgadataflow/annotate_resources.py
index 7b0219d8a8..ee2da2094c 100644
--- a/src/finn/transformation/fpgadataflow/annotate_resources.py
+++ b/src/finn/transformation/fpgadataflow/annotate_resources.py
@@ -78,7 +78,9 @@ def apply(self, model):
                 # recurse into model to manually annotate per-layer resources
                 sdp_model_filename = getCustomOp(node).get_nodeattr("model")
                 sdp_model = ModelWrapper(sdp_model_filename)
-                sdp_model = sdp_model.transform(AnnotateResources(self.mode, self.res_dict))
+                sdp_model = sdp_model.transform(
+                    AnnotateResources(self.mode, self.fpgapart, self.res_dict)
+                )
                 sdp_dict = sdp_model.get_metadata_prop("res_total_" + self.mode)
                 sdp_dict = eval(sdp_dict)
                 # save transformed model
diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py
index 387bf16c95..6fd7cb5e66 100644
--- a/tests/end2end/test_end2end_bnn_pynq.py
+++ b/tests/end2end/test_end2end_bnn_pynq.py
@@ -706,7 +706,7 @@ def test_ipgen(self, topology, wbits, abits, board):
         build_data = get_build_env(board, target_clk_ns)
         if build_data["kind"] == "alveo" and ("VITIS_PATH" not in os.environ):
             pytest.skip("VITIS_PATH not set")
-        prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "fold")
+        prev_chkpt_name = get_checkpoint_name(topology, wbits, abits, "minimize_bit_width")
         model = load_test_checkpoint_or_skip(prev_chkpt_name)
         model = model.transform(GiveUniqueNodeNames())
         model = model.transform(PrepareIP(build_data["part"], target_clk_ns))