diff --git a/src/goodbyedpi.c b/src/goodbyedpi.c
index f9011a5..2ce30d9 100644
--- a/src/goodbyedpi.c
+++ b/src/goodbyedpi.c
@@ -770,7 +770,7 @@ int main(int argc, char *argv[]) {
                                 );
 
                             WinDivertHelperCalcChecksums(
-                                packet, packetLen - packet_dataLen + http_fragment_size, 0
+                                packet, packetLen - packet_dataLen + http_fragment_size, &addr, 0
                             );
                             WinDivertSend(
                                 w_filter, packet,
@@ -975,11 +975,7 @@ int main(int argc, char *argv[]) {
             if (should_reinject) {
                 //printf("Re-injecting!\n");
                 if (should_recalc_checksum) {
-                    WinDivertHelperCalcChecksums(packet, packetLen, 0);
-                }
-                else {
-                    WinDivertHelperCalcChecksums(packet, packetLen,
-                                                 WINDIVERT_HELPER_NO_REPLACE);
+                    WinDivertHelperCalcChecksums(packet, packetLen, &addr, NULL);
                 }
                 WinDivertSend(w_filter, packet, packetLen, &addr, NULL);
             }