struct safexcel_crypto_priv *priv;
 
        u32 alg;
-       u32 digest;
 
        u32 ipad[SHA1_DIGEST_SIZE / sizeof(u32)];
        u32 opad[SHA1_DIGEST_SIZE / sizeof(u32)];
        int nents;
        dma_addr_t result_dma;
 
+       u32 digest;
+
        u8 state_sz;    /* expected sate size, only set once */
        u32 state[SHA256_DIGEST_SIZE / sizeof(u32)] __aligned(sizeof(u32));
 
        u64 len;
        u64 processed;
 
+       u32 digest;
+
        u32 state[SHA256_DIGEST_SIZE / sizeof(u32)];
        u8 cache[SHA256_BLOCK_SIZE];
 };
 
        cdesc->control_data.control0 |= CONTEXT_CONTROL_TYPE_HASH_OUT;
        cdesc->control_data.control0 |= ctx->alg;
-       cdesc->control_data.control0 |= ctx->digest;
+       cdesc->control_data.control0 |= req->digest;
 
-       if (ctx->digest == CONTEXT_CONTROL_DIGEST_PRECOMPUTED) {
+       if (req->digest == CONTEXT_CONTROL_DIGEST_PRECOMPUTED) {
                if (req->processed) {
                        if (ctx->alg == CONTEXT_CONTROL_CRYPTO_ALG_SHA1)
                                cdesc->control_data.control0 |= CONTEXT_CONTROL_SIZE(6);
                        if (req->finish)
                                ctx->base.ctxr->data[i] = cpu_to_le32(req->processed / blocksize);
                }
-       } else if (ctx->digest == CONTEXT_CONTROL_DIGEST_HMAC) {
+       } else if (req->digest == CONTEXT_CONTROL_DIGEST_HMAC) {
                cdesc->control_data.control0 |= CONTEXT_CONTROL_SIZE(10);
 
                memcpy(ctx->base.ctxr->data, ctx->ipad, digestsize);
        if (ctx->base.ctxr) {
                if (priv->version == EIP197 &&
                    !ctx->base.needs_inv && req->processed &&
-                   ctx->digest == CONTEXT_CONTROL_DIGEST_PRECOMPUTED)
+                   req->digest == CONTEXT_CONTROL_DIGEST_PRECOMPUTED)
                        /* We're still setting needs_inv here, even though it is
                         * cleared right away, because the needs_inv flag can be
                         * set in other functions and we want to keep the same
 
 static int safexcel_ahash_update(struct ahash_request *areq)
 {
-       struct safexcel_ahash_ctx *ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(areq));
        struct safexcel_ahash_req *req = ahash_request_ctx(areq);
        struct crypto_ahash *ahash = crypto_ahash_reqtfm(areq);
 
         * We're not doing partial updates when performing an hmac request.
         * Everything will be handled by the final() call.
         */
-       if (ctx->digest == CONTEXT_CONTROL_DIGEST_HMAC)
+       if (req->digest == CONTEXT_CONTROL_DIGEST_HMAC)
                return 0;
 
        if (req->hmac)
        export->len = req->len;
        export->processed = req->processed;
 
+       export->digest = req->digest;
+
        memcpy(export->state, req->state, req->state_sz);
        memcpy(export->cache, req->cache, crypto_ahash_blocksize(ahash));
 
        req->len = export->len;
        req->processed = export->processed;
 
+       req->digest = export->digest;
+
        memcpy(req->cache, export->cache, crypto_ahash_blocksize(ahash));
        memcpy(req->state, export->state, req->state_sz);
 
        req->state[4] = SHA1_H4;
 
        ctx->alg = CONTEXT_CONTROL_CRYPTO_ALG_SHA1;
-       ctx->digest = CONTEXT_CONTROL_DIGEST_PRECOMPUTED;
+       req->digest = CONTEXT_CONTROL_DIGEST_PRECOMPUTED;
        req->state_sz = SHA1_DIGEST_SIZE;
 
        return 0;
 
 static int safexcel_hmac_sha1_init(struct ahash_request *areq)
 {
-       struct safexcel_ahash_ctx *ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(areq));
+       struct safexcel_ahash_req *req = ahash_request_ctx(areq);
 
        safexcel_sha1_init(areq);
-       ctx->digest = CONTEXT_CONTROL_DIGEST_HMAC;
+       req->digest = CONTEXT_CONTROL_DIGEST_HMAC;
        return 0;
 }
 
        req->state[7] = SHA256_H7;
 
        ctx->alg = CONTEXT_CONTROL_CRYPTO_ALG_SHA256;
-       ctx->digest = CONTEXT_CONTROL_DIGEST_PRECOMPUTED;
+       req->digest = CONTEXT_CONTROL_DIGEST_PRECOMPUTED;
        req->state_sz = SHA256_DIGEST_SIZE;
 
        return 0;
        req->state[7] = SHA224_H7;
 
        ctx->alg = CONTEXT_CONTROL_CRYPTO_ALG_SHA224;
-       ctx->digest = CONTEXT_CONTROL_DIGEST_PRECOMPUTED;
+       req->digest = CONTEXT_CONTROL_DIGEST_PRECOMPUTED;
        req->state_sz = SHA256_DIGEST_SIZE;
 
        return 0;