Make test work with python3