diff --git a/extensions/xt_TEE.c b/extensions/xt_TEE.c
index ce25663..ad90cfe 100644
--- a/extensions/xt_TEE.c
+++ b/extensions/xt_TEE.c
@@ -166,11 +166,8 @@ tee_tg4(struct sk_buff **pskb, const struct xt_target_param *par)
 	 * If we are in INPUT, the checksum must be recalculated since
 	 * the length could have changed as a result of defragmentation.
 	 */
-	if (par->hooknum == NF_INET_LOCAL_IN) {
-		struct iphdr *iph = ip_hdr(skb);
-		iph->check = 0;
-		iph->check = ip_fast_csum((unsigned char *)iph, iph->ihl);
-	}
+	if (par->hooknum == NF_INET_LOCAL_IN)
+		ip_send_check(ip_hdr(skb));
 
 	/*
 	 * Copy the skb, and route the copy. Will later return %XT_CONTINUE for