diff --git a/trainer.h b/trainer.h
index 5936ac2321d3439e94cdb3b2fa65d166e9b8cd57..a4a5e11d9fffb0245f16951c2493c7cabdb48363 100644
--- a/trainer.h
+++ b/trainer.h
@@ -279,7 +279,7 @@ class MPI_Trainer: public Trainer {
           descPHI2, dm2.T.ptr(), &ib, &jb, descB2, work2, &lwork2, &info2);
 
       // get weight vector, for context1 
-      pdgemr2d_(&PHI_rows, &ione, dm2.T.ptr(), &ione, &ione, descB2,
+      pdgemr2d_(&PHI_cols, &ione, dm2.T.ptr(), &ione, &ione, descB2,
           dm.T.ptr(), &ione, &ione, descB, &context1);
 
       if (rank==0) {