small fix (Python2 compatibility)