Fixing tests?

Tests are working locally but not in Travis