Bah, missed changes after refactor

Thanks, travis-ci!