Fix test so it runs on Travis