Just fix a bug mentioned at here https://github.com/flyteorg/flyte/issues/3813 Please review it here, https://github.com/flyteorg/flytekit/pull/1793 thanks a lot