Fix unittests for travis-ci