Fixed error with running tests with travis-ci