Fix unittest for travis