diff --git a/flytekit/types/structured/snowflake.py b/flytekit/types/structured/snowflake.py index 5af2a93430..050a77bfb8 100644 --- a/flytekit/types/structured/snowflake.py +++ b/flytekit/types/structured/snowflake.py @@ -24,7 +24,7 @@ def get_private_key(): import flytekit - pk_string = flytekit.current_context().secrets.get(TASK_TYPE, "private_key", encode_mode="rb") + pk_string = flytekit.current_context().secrets.get(SNOWFLAKE, "private_key", encode_mode="rb") p_key = serialization.load_pem_private_key(pk_string, password=None, backend=default_backend()) pkb = p_key.private_bytes(