diff --git a/programs/test/selftest.c b/programs/test/selftest.c
index 333e3a1..0068793 100644
--- a/programs/test/selftest.c
+++ b/programs/test/selftest.c
@@ -86,13 +86,13 @@
         return( ret );
 #endif
 
-#if defined(POLARSSL_SHA2_C)
-    if( ( ret = sha2_self_test( v ) ) != 0 )
+#if defined(POLARSSL_SHA256_C)
+    if( ( ret = sha256_self_test( v ) ) != 0 )
         return( ret );
 #endif
 
-#if defined(POLARSSL_SHA4_C)
-    if( ( ret = sha4_self_test( v ) ) != 0 )
+#if defined(POLARSSL_SHA512_C)
+    if( ( ret = sha512_self_test( v ) ) != 0 )
         return( ret );
 #endif