Fix test on travis

Close #1