Fix test to run on Travis