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