diff --git a/profiler/src/main/java/com/splunk/opentelemetry/profiler/snapshot/TraceRegistry.java b/profiler/src/main/java/com/splunk/opentelemetry/profiler/snapshot/TraceRegistry.java
index 5bf412dde..610f721f1 100644
--- a/profiler/src/main/java/com/splunk/opentelemetry/profiler/snapshot/TraceRegistry.java
+++ b/profiler/src/main/java/com/splunk/opentelemetry/profiler/snapshot/TraceRegistry.java
@@ -17,18 +17,19 @@
 package com.splunk.opentelemetry.profiler.snapshot;
 
 import io.opentelemetry.api.trace.SpanContext;
-import java.util.Map;
+import java.util.Collections;
+import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 
 class TraceRegistry {
-  private final Map<String, SpanContext> traceIds = new ConcurrentHashMap<>();
+  private final Set<String> traceIds = Collections.newSetFromMap(new ConcurrentHashMap<>());
 
   public void register(SpanContext spanContext) {
-    traceIds.put(spanContext.getTraceId(), spanContext);
+    traceIds.add(spanContext.getTraceId());
   }
 
   public boolean isRegistered(SpanContext spanContext) {
-    return traceIds.containsKey(spanContext.getTraceId());
+    return traceIds.contains(spanContext.getTraceId());
   }
 
   public void unregister(SpanContext spanContext) {