Something like this fixes my issue.