diff --git a/python_modules/dagster/dagster/_core/execution/context/input.py b/python_modules/dagster/dagster/_core/execution/context/input.py
index b4babbbad65cd..6b287fd235d2c 100644
--- a/python_modules/dagster/dagster/_core/execution/context/input.py
+++ b/python_modules/dagster/dagster/_core/execution/context/input.py
@@ -405,7 +405,7 @@ def asset_partitions_time_window(self) -> TimeWindow:
                 "Tried to access asset_partitions_time_window, but the asset is not partitioned.",
             )
 
-        return self.step_context.asset_partitions_time_window_for_input(self.name)
+        return self.step_context.asset_partitions_time_window_for_upstream(self.asset_key)
 
     @public
     def get_identifier(self) -> Sequence[str]: