1 files changed, 10 insertions, 10 deletions

diff --git a/drivers/mtd/nand/raw/qcom_nandc.c b/drivers/mtd/nand/raw/qcom_nandc.c
index 46c62a31fa46..920e7375084f 100644
--- a/drivers/mtd/nand/raw/qcom_nandc.c
+++ b/drivers/mtd/nand/raw/qcom_nandc.c
@@ -2833,6 +2833,16 @@ static int qcom_nand_host_init_and_register(struct qcom_nand_controller *nandc,
         if (ret)
                 return ret;
 
+        if (nandc->props->is_bam) {
+                free_bam_transaction(nandc);
+                nandc->bam_txn = alloc_bam_transaction(nandc);
+                if (!nandc->bam_txn) {
+                        dev_err(nandc->dev,
+                                "failed to allocate bam transaction\n");
+                        return -ENOMEM;
+                }
+        }
+
         ret = mtd_device_register(mtd, NULL, 0);
         if (ret)
                 nand_cleanup(chip);
@@ -2847,16 +2857,6 @@ static int qcom_probe_nand_devices(struct qcom_nand_controller *nandc)
         struct qcom_nand_host *host;
         int ret;
 
-        if (nandc->props->is_bam) {
-                free_bam_transaction(nandc);
-                nandc->bam_txn = alloc_bam_transaction(nandc);
-                if (!nandc->bam_txn) {
-                        dev_err(nandc->dev,
-                                "failed to allocate bam transaction\n");
-                        return -ENOMEM;
-                }
-        }
-
         for_each_available_child_of_node(dn, child) {
                 host = devm_kzalloc(dev, sizeof(*host), GFP_KERNEL);
                 if (!host) {