diff --git a/src/main/scala/xiangshan/frontend/BPU.scala b/src/main/scala/xiangshan/frontend/BPU.scala
index ef5410bb4f62f82b3fc60aa4476a90c5f6277b36..831fd7447d876ba6c71043e449e1d561cdeb3ca9 100644
--- a/src/main/scala/xiangshan/frontend/BPU.scala
+++ b/src/main/scala/xiangshan/frontend/BPU.scala
@@ -329,10 +329,10 @@ class BPU extends BaseBPU {
   ubtb.io.inMask := s1_inLatch.bits.inMask
 
   // Wrap ubtb response into resp_in and brInfo_in
-  s1_resp_in.ubtb <> ubtb.io.resp
+  s1_resp_in.ubtb <> ubtb.io.out
   for (i <- 0 until PredictWidth) {
-    s1_brInfo_in(i).ubtbWriteWay := ubtb.io.meta.writeWay(i)
-    s1_brInfo_in(i).ubtbHits := ubtb.io.out.targets(i).valid
+    s1_brInfo_in(i).ubtbWriteWay := ubtb.io.uBTBBranchInfo.writeWay(i)
+    s1_brInfo_in(i).ubtbHits := ubtb.io.uBTBBranchInfo.hits(i)
   }
 
   btb.io.flush := io.flush(0) // TODO: fix this