diff --git a/api/apiIntegration_test.go b/api/apiIntegration_test.go
index 26057836710c46decca5a815242459d3ad8d058a..9839da725ee9a573f3afc768654ea40e126ddd7f 100644
--- a/api/apiIntegration_test.go
+++ b/api/apiIntegration_test.go
@@ -55,18 +55,18 @@ func testSetupIntegration() {
 	}
 
 	gnmiMessages = map[string]pb.Message{
-		"../proto/cap-resp-arista-ceos":                  &gpb.CapabilityResponse{},
-		"../proto/req-full-node":                         &gpb.GetRequest{},
-		"../proto/req-full-node-arista-ceos":             &gpb.GetRequest{},
-		"../proto/req-interfaces-arista-ceos":            &gpb.GetRequest{},
-		"../proto/req-interfaces-interface-arista-ceos":  &gpb.GetRequest{},
-		"../proto/req-interfaces-wildcard":               &gpb.GetRequest{},
-		"../proto/resp-full-node":                        &gpb.GetResponse{},
-		"../proto/resp-full-node-arista-ceos":            &gpb.GetResponse{},
-		"../proto/resp-interfaces-arista-ceos":           &gpb.GetResponse{},
-		"../proto/resp-interfaces-interface-arista-ceos": &gpb.GetResponse{},
-		"../proto/resp-interfaces-wildcard":              &gpb.GetResponse{},
-		"../proto/resp-set-system-config-hostname":       &gpb.SetResponse{},
+		"../test/proto/cap-resp-arista-ceos":                  &gpb.CapabilityResponse{},
+		"../test/proto/req-full-node":                         &gpb.GetRequest{},
+		"../test/proto/req-full-node-arista-ceos":             &gpb.GetRequest{},
+		"../test/proto/req-interfaces-arista-ceos":            &gpb.GetRequest{},
+		"../test/proto/req-interfaces-interface-arista-ceos":  &gpb.GetRequest{},
+		"../test/proto/req-interfaces-wildcard":               &gpb.GetRequest{},
+		"../test/proto/resp-full-node":                        &gpb.GetResponse{},
+		"../test/proto/resp-full-node-arista-ceos":            &gpb.GetResponse{},
+		"../test/proto/resp-interfaces-arista-ceos":           &gpb.GetResponse{},
+		"../test/proto/resp-interfaces-interface-arista-ceos": &gpb.GetResponse{},
+		"../test/proto/resp-interfaces-wildcard":              &gpb.GetResponse{},
+		"../test/proto/resp-set-system-config-hostname":       &gpb.SetResponse{},
 	}
 	for k, v := range gnmiMessages {
 		if err := proto.Read(k, v); err != nil {