diff --git a/test/test_class1_allele_specific_cv_and_train_command.py b/test/test_class1_allele_specific_cv_and_train_command.py
index 9f0c193bb2330ecfb0aa6438fd67cd67d61d43a6..99d1a991139914da6d823ae968525d1303081709 100644
--- a/test/test_class1_allele_specific_cv_and_train_command.py
+++ b/test/test_class1_allele_specific_cv_and_train_command.py
@@ -13,6 +13,12 @@ from mhcflurry.class1_allele_specific import cv_and_train_command
 from mhcflurry import downloads, predict
 from mhcflurry.class1_allele_specific.train import HYPERPARAMETER_DEFAULTS
 
+try:
+    import kubeface
+    KUBEFACE_INSTALLED = True
+except ImportError:
+    KUBEFACE_INSTALLED = False
+
 
 def test_small_run():
     base_temp_dir = tempfile.mkdtemp()
@@ -61,8 +67,9 @@ def test_small_run():
         "--alleles", "HLA-A0201", "HLA-A0301",
         "--verbose",
         "--num-local-threads", "1",
-        "--storage-prefix", "/tmp/",
     ]
+    if KUBEFACE_INSTALLED:
+        args.extend(["--storage-prefix", "/tmp/"])
     print("Running cv_and_train_command with args: %s " % str(args))
 
     cv_and_train_command.run(args)