Fix issue caused by import syntax in some versions of Python