Minor change for py3 compatibility