diff --git a/training_config/nli/snli_roberta.jsonnet b/training_config/nli/snli_roberta.jsonnet
index 9c03c9279..17969ee6d 100644
--- a/training_config/nli/snli_roberta.jsonnet
+++ b/training_config/nli/snli_roberta.jsonnet
@@ -42,7 +42,8 @@ local cls_is_last_token = false;
       "hidden_dims": transformer_dim,
       "activations": "tanh"
     },
-    "dropout": 0.1
+    "dropout": 0.1,
+    "namespace": "tags"
   },
   "data_loader": {
     "batch_sampler": {