Fix to make tests work on Travis CI