Fix get parent element

https://next.getflow.com/workspaces/1/tasks/5617983