index 0471cae..afb91d7 100644
--- a/drivers/net/ehea/ehea_main.c
+++ b/drivers/net/ehea/ehea_main.c
@@ -533,8 +533,20 @@ static inline void ehea_fill_skb(struct net_device *dev,
        int length = cqe->num_bytes_transfered - 4;     /*remove CRC */

        skb_put(skb, length);
-       skb->ip_summed = CHECKSUM_UNNECESSARY;
        skb->protocol = eth_type_trans(skb, dev);
+       printk("Protocol %d\n", skb->protocol);
+       printk("%d\n", ip_hdr(skb)->protocol);
+       if (skb->protocol == htons(ETH_P_IP) &&
+           (ip_hdr(skb)->protocol == IPPROTO_TCP ||
+            ip_hdr(skb)->protocol == IPPROTO_UDP)){
+               skb->ip_summed = CHECKSUM_UNNECESSARY;
+               printk("Unnescessary\n");
+       }
+       else{
+               printk("Complete\n");
+               skb->ip_summed = CHECKSUM_COMPLETE;
+               skb->csum = cqe->inet_checksum_value;
+       }
 }