Use flex
Merge pull request #2 from Versal/fix/flex

Use flex