fix(tests): add py3 compatibility