Fix bug in travis build