Another test fix for TravisCI