int rc;
 
        /* Allocate "this" context */
-       drvdata->ivgen_handle = kzalloc(sizeof(*drvdata->ivgen_handle),
-                                       GFP_KERNEL);
-       if (!drvdata->ivgen_handle)
+       ivgen_ctx = kzalloc(sizeof(*ivgen_ctx), GFP_KERNEL);
+       if (!ivgen_ctx)
                return -ENOMEM;
 
-       ivgen_ctx = drvdata->ivgen_handle;
-
        /* Allocate pool's header for initial enc. key/IV */
        ivgen_ctx->pool_meta = dma_alloc_coherent(device, CC_IVPOOL_META_SIZE,
                                                  &ivgen_ctx->pool_meta_dma,
                goto out;
        }
 
+       drvdata->ivgen_handle = ivgen_ctx;
+
        return cc_init_iv_sram(drvdata);
 
 out:
 
  */
 int cc_sram_mgr_init(struct cc_drvdata *drvdata)
 {
+       struct cc_sram_ctx *ctx;
+
        /* Allocate "this" context */
-       drvdata->sram_mgr_handle = kzalloc(sizeof(*drvdata->sram_mgr_handle),
-                                          GFP_KERNEL);
+       ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
 
-       if (!drvdata->sram_mgr_handle)
+       if (!ctx)
                return -ENOMEM;
 
+       drvdata->sram_mgr_handle = ctx;
+
        return 0;
 }