fix: Tests failed in Python3