Attempt to fix Python 3 error