Ugly hack for Python2 compatibility :)