Fix bug with test assertion in travis-ci