Edited tests for TRAVIS CI