1 files changed, 3 insertions, 2 deletions

diff --git a/arch/s390/crypto/sha.h b/arch/s390/crypto/sha.h
index b7a52ab5db97..1ceafa571eab 100644
--- a/arch/s390/crypto/sha.h
+++ b/arch/s390/crypto/sha.h
@@ -19,11 +19,12 @@
 #include <crypto/sha.h>
 
 /* must be big enough for the largest SHA variant */
-#define SHA_MAX_BLOCK_SIZE      SHA256_BLOCK_SIZE
+#define SHA_MAX_STATE_SIZE      16
+#define SHA_MAX_BLOCK_SIZE      SHA512_BLOCK_SIZE
 
 struct s390_sha_ctx {
         u64 count;              /* message length in bytes */
-        u32 state[8];
+        u32 state[SHA_MAX_STATE_SIZE];
         u8 buf[2 * SHA_MAX_BLOCK_SIZE];
         int func;               /* KIMD function to use */
 };