diff --git a/unittest/vslib/test_sai_vs_lag.cpp b/unittest/vslib/test_sai_vs_lag.cpp
index d858483e6c8f..a185f8897129 100644
--- a/unittest/vslib/test_sai_vs_lag.cpp
+++ b/unittest/vslib/test_sai_vs_lag.cpp
@@ -10,7 +10,7 @@ TEST(libsaivs, lag)
 {
     sai_lag_api_t *api = nullptr;
 
-    sai_api_query(SAI_API_VLAN, (void**)&api);
+    sai_api_query(SAI_API_LAG, (void**)&api);
 
     EXPECT_NE(api, nullptr);