Make tests work with python3