diff --git a/scripts/run_tests_locally-nbi.sh b/scripts/run_tests_locally-nbi-ietf-l2vpn.sh
similarity index 96%
rename from scripts/run_tests_locally-nbi.sh
rename to scripts/run_tests_locally-nbi-ietf-l2vpn.sh
index 1be0451588195cf6d97c7742e7d2c7f29e08441a..f6b53e432122e007a8f4c99150ab5d090e84dca3 100755
--- a/scripts/run_tests_locally-nbi.sh
+++ b/scripts/run_tests_locally-nbi-ietf-l2vpn.sh
@@ -21,4 +21,4 @@ RCFILE=$PROJECTDIR/coverage/.coveragerc
 
 # Run unitary tests and analyze coverage of code at same time
 coverage run --rcfile=$RCFILE --append -m pytest --log-level=INFO --verbose \
-    nbi/tests/test_unitary.py
+    nbi/tests/test_ietf_l2vpn.py
diff --git a/src/nbi/tests/test_ietf_network.py b/src/nbi/tests/test_ietf_network.py
index 1977bd50854553751988e7161589dada53ff25a3..78183c3daec0c4d96d6eebdaf7dfa44680f19fa6 100644
--- a/src/nbi/tests/test_ietf_network.py
+++ b/src/nbi/tests/test_ietf_network.py
@@ -15,6 +15,7 @@
 from typing import Dict
 import deepdiff, json, logging, operator
 from common.Constants import DEFAULT_CONTEXT_NAME
+from common.proto.context_pb2 import ContextId
 from common.tools.descriptor.Loader import DescriptorLoader, check_descriptor_load_results, validate_empty_scenario
 from common.tools.object_factory.Context import json_context_id
 from context.client.ContextClient import ContextClient