diff --git a/python_modules/dagster/dagster/_core/execution/plan/compute.py b/python_modules/dagster/dagster/_core/execution/plan/compute.py
index 7bb4c8d677b43..d48ee9b1363b9 100644
--- a/python_modules/dagster/dagster/_core/execution/plan/compute.py
+++ b/python_modules/dagster/dagster/_core/execution/plan/compute.py
@@ -62,7 +62,7 @@
 
 
 def _get_op_context(
-    context: Union[OpExecutionContext, AssetExecutionContext]
+    context: Union[OpExecutionContext, AssetExecutionContext],
 ) -> OpExecutionContext:
     if isinstance(context, AssetExecutionContext):
         return context.op_execution_context
diff --git a/python_modules/dagster/dagster/_core/execution/plan/compute_generator.py b/python_modules/dagster/dagster/_core/execution/plan/compute_generator.py
index dd930fa0c98a6..21583e20760e9 100644
--- a/python_modules/dagster/dagster/_core/execution/plan/compute_generator.py
+++ b/python_modules/dagster/dagster/_core/execution/plan/compute_generator.py
@@ -40,7 +40,7 @@
 
 
 def _get_op_context(
-    context: Union[OpExecutionContext, AssetExecutionContext]
+    context: Union[OpExecutionContext, AssetExecutionContext],
 ) -> OpExecutionContext:
     if isinstance(context, AssetExecutionContext):
         return context.op_execution_context