Updates for unit tests on travis