diff --git a/nucleus/integration_test.go b/nucleus/integration_test.go
index 171cfd3b9570067509afe4e7007858825d56f979..455138f97b17d93667b8e1e888bcaafe586ea350 100644
--- a/nucleus/integration_test.go
+++ b/nucleus/integration_test.go
@@ -263,6 +263,7 @@ func TestGnmi_SubscribeIntegration(t *testing.T) {
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
+			var wantErr = tt.wantErr
 			g, err := NewGnmiTransport(tt.fields.opt)
 			if err != nil {
 				t.Error(err)
@@ -271,10 +272,10 @@ func TestGnmi_SubscribeIntegration(t *testing.T) {
 			ctx := context.WithValue(context.Background(), CtxKeyOpts, tt.args.opts) //nolint
 			ctx, cancel := context.WithCancel(ctx)
 			go func() {
-				err = g.Subscribe(ctx)
-				if (err != nil) != tt.wantErr {
-					if !tt.wantErr {
-						if err.Error() != "rpc error: code = Canceled desc = context canceled" {
+				subErr := g.Subscribe(ctx)
+				if (subErr != nil) != wantErr {
+					if !wantErr && subErr != nil{
+						if subErr.Error() != "rpc error: code = Canceled desc = context canceled" {
 							t.Errorf("Subscribe() error = %v, wantErr %v", err, tt.wantErr)
 						}
 					}