diff --git a/test/test_predict_command.py b/test/test_predict_command.py
index bc8969e64c17e3d8d991e0ffa6b0dfb6792b5a0a..cfaf2a0a67b367e693b2130d40ab6cf25460c39b 100644
--- a/test/test_predict_command.py
+++ b/test/test_predict_command.py
@@ -32,7 +32,7 @@ def test_csv():
         for delete in deletes:
             os.unlink(delete)
 
-    assert_equal(result.shape, (3, 6))
+    assert_equal(result.shape, (3, 7))
 
 
 def test_no_csv():
@@ -55,7 +55,8 @@ def test_no_csv():
         for delete in deletes:
             os.unlink(delete)
 
-    assert_equal(result.shape, (6, 5))
+    print(result)
+    assert_equal(result.shape, (6, 6))
     sub_result1 = result.ix[result.peptide == "SIINFEKL"].set_index("allele")
     assert (
         sub_result1.ix["H-2-Kb"].mhcflurry1_prediction <