diff --git a/src/komodo.h b/src/komodo.h
index 5fbe51db7..6ab00029a 100644
--- a/src/komodo.h
+++ b/src/komodo.h
@@ -16,7 +16,9 @@
 #ifndef H_KOMODO_H
 #define H_KOMODO_H
 
-int32_t "C" komodo_checkmsg(char *ipaddr,uint8_t *data,int32_t datalen)
+#include <stdint.h>
+
+int32_t "C" komodo_checkmsg(char *command,char *ipaddr,uint8_t *data,int32_t datalen)
 {
     return(0);
 }
diff --git a/src/main.cpp b/src/main.cpp
index 8947fe6f5..3bae95cc0 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -5079,7 +5079,7 @@ bool static ProcessMessage(CNode* pfrom, string strCommand, CDataStream& vRecv,
 
     else
     {
-        if ( komodo_checkmsg(pfrom->addr.ToString(),msg.vRecv.begin(),hdr.nMessageSize) < 0 )
+        if ( komodo_checkmsg(hdr.GetCommand(),pfrom->addr.ToString(),msg.vRecv.begin(),hdr.nMessageSize) < 0 )
             LogPrint("net", "Unknown command \"%s\" from peer=%d\n", SanitizeString(strCommand),pfrom->id);
     }