Fixed py3 version of tests