diff --git a/mhcflurry/class1_ligandome_predictor.py b/mhcflurry/class1_ligandome_predictor.py
index 74a9e1db4715657448e3cfe4d1fc6f8eb5c139f2..2dc4350932844c8f3e7c395397fd21c0f438e7ec 100644
--- a/mhcflurry/class1_ligandome_predictor.py
+++ b/mhcflurry/class1_ligandome_predictor.py
@@ -156,8 +156,8 @@ class Class1LigandomePredictor(object):
         y_true = tf.reshape(tf.cast(y_true, tf.bool), (-1,))
 
         pos = tf.boolean_mask(y_pred, y_true)
-        #pos_max = tf.reduce_max(pos, axis=1)
-        
+        pos_max = tf.reduce_max(pos, axis=1)
+
         neg = tf.boolean_mask(y_pred, tf.logical_not(y_true))
         result = tf.reduce_sum(
             tf.maximum(0.0, tf.reshape(neg, (-1, 1)) - pos_max + delta) ** 2)