[fix] Use the correct api reference