Quick change to get things working on Travis CI