diff --git a/.travis.yml b/.travis.yml
index d1ec644f7812ee2ca54711e519555453e1bbe8c6..696dbd599b60da0596ede257c9b3e6012cf03c69 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -17,6 +17,7 @@ before_install:
   - conda update -q conda
   # Useful for debugging any issues with conda
   - conda info -a
+  - free -m
 addons:
   apt:
     packages:
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 86a0cdf5a11766e02116e7c27ee9020a6b9fdfb3..48db82f960d61ec1e5eb7fe9a2c47972cf26bbb9 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -1368,7 +1368,10 @@ class Class1NeuralNetwork(object):
                   m is the length of the vectors used to represent amino acids
         """
         from keras.models import clone_model
+        import tensorflow as tf
         import keras.backend as K
+        K.get_session().run(tf.global_variables_initializer())
+
         reshaped = allele_representations.reshape(
             (allele_representations.shape[0], -1))
         original_model = self.network()