New file on File Open cannot resolve template for Episodes

Need help here, I’ve been following the guide for episodic workflow from this link:

but somehow when I tried to create new file from the File Open dialog it returns this error:

Looks like maybe the shot entity doesn’t have ‘Episode’ entity. How can I setup the ‘Episode’ entity for the shot_step? If I create a new field for ‘Shot’ entity while I already setup the ‘Episode’ entity from the ‘Sequence’, wouldn’t it be doubled entry and not very effective? Appreciate any help. Thanks!

Nvm, found a solution.

Anyone having the same issue just need to modify the Episode key in templates.yml to get the episode from the sequence:

type: str
shotgun_entity_type: Shot
shotgun_field_name: sg_sequence.Sequence.episode