diff --git a/doc/requirements.txt b/doc/requirements.txt
index 614ed5144a..e844537584 100644
--- a/doc/requirements.txt
+++ b/doc/requirements.txt
@@ -1,4 +1,4 @@
-sphinx==3.1.1
+sphinx==3.1.2
 sphinx-rtd-theme==0.5.0
 docutils==0.15.2
 packaging==20.9
diff --git a/src/sagemaker/workflow/pipeline_context.py b/src/sagemaker/workflow/pipeline_context.py
index 668923a9ff..e98ca3b244 100644
--- a/src/sagemaker/workflow/pipeline_context.py
+++ b/src/sagemaker/workflow/pipeline_context.py
@@ -16,6 +16,7 @@
 import warnings
 import inspect
 from typing import Dict
+from functools import wraps
 
 from sagemaker.session import Session, SessionSettings
 
@@ -132,6 +133,7 @@ def runnable_by_pipeline(run_func):
     The job will be started during pipeline execution.
     """
 
+    @wraps(run_func)
     def wrapper(*args, **kwargs):
         self_instance = args[0]
         if isinstance(self_instance.sagemaker_session, PipelineSession):