Imp. Fix to run test on Travis CI