Minor change for python 2/3 compat