Try fix tests for python 2.