1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/crypto/caam/caamalg.c b/drivers/crypto/caam/caamalg.c
index 86a0d415b9a7..ad9781e646c0 100644
--- a/drivers/crypto/caam/caamalg.c
+++ b/drivers/crypto/caam/caamalg.c
@@ -2071,13 +2071,15 @@ static int caam_cra_init(struct crypto_tfm *tfm)
                  container_of(alg, struct caam_crypto_alg, crypto_alg);
         struct caam_ctx *ctx = crypto_tfm_ctx(tfm);
         struct caam_drv_private *priv = dev_get_drvdata(caam_alg->ctrldev);
+        struct platform_device *pdev;
         int tgt_jr = atomic_inc_return(&priv->tfm_count);
 
         /*
          * distribute tfms across job rings to ensure in-order
          * crypto request processing per tfm
          */
-        ctx->jrdev = priv->jrdev[(tgt_jr / 2) % priv->total_jobrs];
+        pdev = priv->jrpdev[(tgt_jr / 2) % priv->total_jobrs];
+        ctx->jrdev = &pdev->dev;
 
         /* copy descriptor header template value */
         ctx->class1_alg_type = OP_TYPE_CLASS1_ALG | caam_alg->class1_alg_type;