diff --git a/uncertainty_baselines/datasets/toxic_comments.py b/uncertainty_baselines/datasets/toxic_comments.py
index 7d3a0805b..6cb5f9acd 100644
--- a/uncertainty_baselines/datasets/toxic_comments.py
+++ b/uncertainty_baselines/datasets/toxic_comments.py
@@ -60,7 +60,7 @@
 _IS_TRAIN_NAME = 'is_train'
 _SIGNAL_NAMES = [
     _IS_TRAIN_NAME, 'pseudo_labels', 'use_pseudo_label', 'bias_rank',
-    'uncertainty', 'bias', 'noise', 'variance', 'error'
+    'uncertainty', 'bias', 'noise', 'variance', 'error', 'error_rank',
 ]