diff --git a/src/tests/tools/mock_nbi_dependencies/Dockerfile b/src/tests/tools/mock_nbi_dependencies/Dockerfile
index 2e0bc7b5b1e5c98bdba30224b4d84e85a123472b..3a3a5e854e56e4801e4091ba3d4128d5a0fb08ae 100644
--- a/src/tests/tools/mock_nbi_dependencies/Dockerfile
+++ b/src/tests/tools/mock_nbi_dependencies/Dockerfile
@@ -61,7 +61,8 @@ RUN pip-compile --quiet --output-file=requirements.txt requirements.in
 RUN python3 -m pip install -r requirements.txt
 
 # Add component files into working directory
-COPY src/tests/tools/mock_nbi_dependencies/. ./
+WORKDIR /var/teraflow
+COPY src/tests/tools/mock_nbi_dependencies/. mock_nbi_dependencies/
 
 # Start the service
-ENTRYPOINT ["python", "MockNbiDependencies.py"]
+ENTRYPOINT ["python", "-m", "tests.tools.mock_nbi_dependencies"]
diff --git a/src/tests/tools/mock_nbi_dependencies/README.md b/src/tests/tools/mock_nbi_dependencies/README.md
index 0f3228b1c250f4def0238dcf2e8da0bed7100a7d..a6d69fdf9d2cba84d478ac862b45b100cf1da4bb 100644
--- a/src/tests/tools/mock_nbi_dependencies/README.md
+++ b/src/tests/tools/mock_nbi_dependencies/README.md
@@ -17,5 +17,5 @@ pip install -r src/tests/tools/mock_nbi_dependencies/requirements.in
 ## 2. Run the Mock NBI Dependencies
 Run the Mock NBI Dependencies as follows:
 ```bash
-python src/tests/tools/mock_nbi_dependencies/MockNbiDependencies.py
+PYTHONPATH=./src python -m tests.tools.mock_nbi_dependencies
 ```
diff --git a/src/tests/tools/mock_nbi_dependencies/MockNbiDependencies.py b/src/tests/tools/mock_nbi_dependencies/__main__.py
similarity index 100%
rename from src/tests/tools/mock_nbi_dependencies/MockNbiDependencies.py
rename to src/tests/tools/mock_nbi_dependencies/__main__.py