Fixes unit tests for travis