diff --git a/plugins/dmarc b/plugins/dmarc
index b0437ce..d37a9b4 100644
--- a/plugins/dmarc
+++ b/plugins/dmarc
@@ -192,12 +192,12 @@ sub check_dmarc {
 
     $self->adjust_karma(-3);
     # Add a mark now so the data_post hook can do the real reject
-    $self->connection->notes('reject_dmarc', '1');
+    $transaction->notes('reject_dmarc', '1');
 }
 
 sub reject_dmarc {
     my ($self, $transaction) = @_;
     return $self->get_reject("failed DMARC policy")
-        if ($self->connection->notes('reject_dmarc'));
+        if ($transaction->notes('reject_dmarc'));
     return DECLINED;
 }