diff --git a/dan/datasets/extract/extract.py b/dan/datasets/extract/extract.py
index e3c7603430811231b99e6ac7a84e207e02876d37..59ec9139627cc8cf20c01602e25bba900918158f 100644
--- a/dan/datasets/extract/extract.py
+++ b/dan/datasets/extract/extract.py
@@ -281,7 +281,7 @@ class ArkindexExtractor:
         """
         return " ".join(
             [
-                self.mapping.encode[token] if token in self.mapping else token
+                self.mapping.encode[token] if token in self.mapping.encode else token
                 for token in list(text.strip())
             ]
         )