diff --git a/library/md_wrap.c b/library/md_wrap.c
index 2621d22..d72852b 100644
--- a/library/md_wrap.c
+++ b/library/md_wrap.c
@@ -50,11 +50,11 @@
 #endif
 
 #if defined(POLARSSL_SHA256_C)
-#include "polarssl/sha2.h"
+#include "polarssl/sha256.h"
 #endif
 
 #if defined(POLARSSL_SHA512_C)
-#include "polarssl/sha4.h"
+#include "polarssl/sha512.h"
 #endif
 
 #include <stdlib.h>