Fix unittest for python 3