Try to make it work with python3