Fixed error in test files for travis-ci