1 files changed, 1 insertions, 6 deletions
diff --git a/src/resolve/resolved-dns-stub.c b/src/resolve/resolved-dns-stub.c
index ce994a7ee0..d9e6218081 100644
--- a/src/resolve/resolved-dns-stub.c
+++ b/src/resolve/resolved-dns-stub.c
@@ -348,16 +348,11 @@ static void dns_stub_process_query(Manager *m, DnsStream *s, DnsPacket *p) {
                 /* Remember which queries belong to this stream, so that we can cancel them when the stream
                  * is disconnected early */
 
-                r = set_ensure_allocated(&s->queries, &trivial_hash_ops);
+                r = set_ensure_put(&s->queries, &trivial_hash_ops, q);
                 if (r < 0) {
                         log_oom();
                         goto fail;
                 }
-
-                if (set_put(s->queries, q) < 0) {
-                        log_oom();
-                        goto fail;
-                }
         }
 
         r = dns_query_go(q);