diff --git a/tests/tbf/TbfTest.cpp b/tests/tbf/TbfTest.cpp
index 159c859..39f57ad 100644
--- a/tests/tbf/TbfTest.cpp
+++ b/tests/tbf/TbfTest.cpp
@@ -526,7 +526,7 @@
 
 	the_bts.rcv_rach(0x03, fn, qta);
 
-	ul_tbf = the_bts.ul_tbf_by_tfi(tfi, trx_no);
+	ul_tbf = the_bts.ul_tbf_by_tfi(tfi, trx_no, ts_no);
 	OSMO_ASSERT(ul_tbf != NULL);
 
 	fprintf(stderr, "Got '%s', TA=%d\n",
@@ -605,7 +605,7 @@
 	pdch->rcv_block(&buf[0], num_bytes, 2654270, &meas);
 
 	/* check the TBF */
-	ul_tbf = the_bts.ul_tbf_by_tfi(tfi, trx_no);
+	ul_tbf = the_bts.ul_tbf_by_tfi(tfi, trx_no, ts_no);
 	OSMO_ASSERT(ul_tbf != NULL);
 
 	fprintf(stderr, "Got '%s', TA=%d, CS=%d\n",