Fix for running unit test on travis