diff --git a/tests/alloc/AllocTest.cpp b/tests/alloc/AllocTest.cpp
index 281ac65..5790ade 100644
--- a/tests/alloc/AllocTest.cpp
+++ b/tests/alloc/AllocTest.cpp
@@ -62,13 +62,13 @@
 	for (int i = 0; i < count; ++i) {
 		struct gprs_rlcmac_tbf *tbf;
 
-		tfi = tfi_find_free(bts, dir, &used_trx, 0);
+		tfi = the_bts.tfi_find_free(dir, &used_trx, 0);
 		OSMO_ASSERT(tfi >= 0);
 		tbfs[i] = tbf_alloc(bts, NULL, dir, tfi, used_trx, 0, 0);
 	}
 
 	/* Now check that there are still some TFIs */
-	tfi = tfi_find_free(bts, dir, &used_trx, 0);
+	tfi = the_bts.tfi_find_free(dir, &used_trx, 0);
 	switch (dir) {
 	case GPRS_RLCMAC_UL_TBF:
 		OSMO_ASSERT(tfi >= 0);
@@ -83,7 +83,7 @@
 		if (tbfs[i])
 			tbf_free(tbfs[i]);
 
-	tfi = tfi_find_free(bts, dir, &used_trx, 0);
+	tfi = the_bts.tfi_find_free(dir, &used_trx, 0);
 	OSMO_ASSERT(tfi >= 0);
 
 	tbfs[tfi] = tbf_alloc(bts, NULL, dir, tfi, used_trx, 0, 0);