Fix test because of the python3 data type differences.