Fixed PY2.6 error by changing tests