Fix tests
fix travis ci build