From bb2bc952470b2d69eb3592839aa3ee4f8a43ca92 Mon Sep 17 00:00:00 2001
From: Ismael Gomez <ismagom@gmail.com>
Date: Tue, 31 Oct 2017 22:46:57 -0700
Subject: [PATCH] Fix diversity tx not working for 4 ports

---
 lib/src/phy/phch/pdsch.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/src/phy/phch/pdsch.c b/lib/src/phy/phch/pdsch.c
index 7c744f552..b984adee3 100644
--- a/lib/src/phy/phch/pdsch.c
+++ b/lib/src/phy/phch/pdsch.c
@@ -493,7 +493,7 @@ int srslte_pdsch_cfg_mimo(srslte_pdsch_cfg_t *cfg, srslte_cell_t cell, srslte_ra
           ERROR("Wrong number of transport blocks (%d) for transmit diversity.", nof_tb);
           return SRSLTE_ERROR;
         }
-        cfg->nof_layers = 2;
+        cfg->nof_layers = cell.nof_ports;
         break;
       case SRSLTE_MIMO_TYPE_SPATIAL_MULTIPLEX:
         if (nof_tb == 1) {