Python 3 compatibility for tests