Fix type issue, but using old syntax
This PR is the same as !158 (merged) but using the older syntax https://docs.python.org/3.7/library/typing.html#typing.Union
This PR is the same as !158 (merged) but using the older syntax https://docs.python.org/3.7/library/typing.html#typing.Union