diff --git a/vault/token_store_test.go b/vault/token_store_test.go
index 8b414fa6a6a7..e7022e7c6e87 100644
--- a/vault/token_store_test.go
+++ b/vault/token_store_test.go
@@ -904,7 +904,8 @@ func buildTokenTree(t testing.TB, ts *TokenStore, depth uint64) (root *TokenEntr
 }
 
 func TestTokenStore_RevokeSelf(t *testing.T) {
-	_, ts, _, _ := TestCoreWithTokenStore(t)
+	exp := mockExpiration(t)
+	ts := exp.tokenStore
 
 	ent1 := &TokenEntry{}
 	if err := ts.create(context.Background(), ent1); err != nil {