Fix cli tests on python 3