diff --git a/mhcflurry/affinity_measurement_dataset.py b/mhcflurry/affinity_measurement_dataset.py
index 757e51ed7895924a7f040873f43b7f76fb09b52b..724442193342e2bdd031edf86760acd148399f2f 100644
--- a/mhcflurry/affinity_measurement_dataset.py
+++ b/mhcflurry/affinity_measurement_dataset.py
@@ -757,7 +757,7 @@ class AffinityMeasurementDataset(object):
             (k in keys_to_remove_set)
             for k in zip(self.alleles, self.peptides)
         ])
-        keep_mask = ~remove_mask
+        keep_mask = remove_mask == False
         return self.slice(keep_mask)
 
     def difference(self, other_dataset):