diff --git a/.travis.yml b/.travis.yml
index db7d5c238d6fc16097afae103fb377a2f9955a78..6164353aa6e92ab038441b4e943b744d48d43553 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,3 +1,4 @@
+dist: bionic
 language: python
 python:
   - "3.6"
@@ -17,6 +18,7 @@ before_install:
   # Useful for debugging any issues with conda
   - conda info -a
   - free -m
+  - make -v
 addons:
   apt:
     packages:
diff --git a/docs/commandline_tutorial.rst b/docs/commandline_tutorial.rst
index 874f68606fb29f138e966345c3f07105cd1ca991..a2a6d22e505c09c0bc39e569bfb4904af22c9c1c 100644
--- a/docs/commandline_tutorial.rst
+++ b/docs/commandline_tutorial.rst
@@ -161,7 +161,7 @@ training data. The data we use for our released predictors can be downloaded wit
 It looks like this:
 
 .. command-output::
-    bzcat "$(mhcflurry-downloads path data_curated)/curated_training_data.no_mass_spec.csv.bz2" | head -n 3
+    bzcat "$(mhcflurry-downloads path data_curated)/curated_training_data.csv.bz2" | head -n 3
     :shell:
     :nostderr:
 
diff --git a/test/test_class1_presentation_predictor.py b/test/test_class1_presentation_predictor.py
index 35a9c6dce11e277a7153d5b234fc2f716bed8799..e144c1fd0303e94c7b0544ec48c957d55aef3039 100644
--- a/test/test_class1_presentation_predictor.py
+++ b/test/test_class1_presentation_predictor.py
@@ -91,13 +91,13 @@ def test_basic():
             alleles=experiment_to_alleles,
             n_flanks=test_df.n_flank.values,
             c_flanks=test_df.c_flank.values,
-            verbose=2)
+            verbose=2).presentation_score.values
 
         test_df["prediction2"] = predictor.predict(
             peptides=test_df.peptide.values,
             sample_names=test_df.sample_id.values,
             alleles=experiment_to_alleles,
-            verbose=2)
+            verbose=2).presentation_score.values
 
     add_prediction_cols(test_df, predictor)