Fix tests & travis-ci