Fix travis
Fix tests in travis