diff --git a/torch_modules/src/DistanceModule.cpp b/torch_modules/src/DistanceModule.cpp
index f529537bd29848f8063ad226706e0003db11eb7f..1b8034b51e2b19ec65294e5ae8f2bc7d50cfa016 100644
--- a/torch_modules/src/DistanceModule.cpp
+++ b/torch_modules/src/DistanceModule.cpp
@@ -100,7 +100,7 @@ void DistanceModuleImpl::addToContext(std::vector<std::vector<long>> & context,
         long dist = std::abs(config.getRelativeDistance(from, to));
 
         if (dist <= threshold)
-          contextElement.emplace_back(dict.getIndexOrInsert(fmt::format("{}", dist)));
+          contextElement.emplace_back(dict.getIndexOrInsert(fmt::format("distance({})", dist)));
         else
           contextElement.emplace_back(dict.getIndexOrInsert(Dict::unknownValueStr));
       }