fix python 2 failure