diff --git a/mhcflurry/testing_utils.py b/mhcflurry/testing_utils.py
index 8c14fa15e36c189f9175b7b8070908accb544376..f6cc079b375acbad60ddac2e7974ac72377b06fe 100644
--- a/mhcflurry/testing_utils.py
+++ b/mhcflurry/testing_utils.py
@@ -3,5 +3,5 @@ from . import Class1NeuralNetwork
 
 def module_cleanup():
     import keras.backend as K
-    Class1NeuralNetwork.KERAS_MODELS_CACHE.clear()
+    Class1NeuralNetwork.clear_model_cache()
     K.clear_session()