diff --git a/src/AFQMC/Utilities/taskgroup.h b/src/AFQMC/Utilities/taskgroup.h
index 384599ebf..fba8cfc0f 100644
--- a/src/AFQMC/Utilities/taskgroup.h
+++ b/src/AFQMC/Utilities/taskgroup.h
@@ -411,7 +411,7 @@ private:
       ncclUniqueId id;
       if (tgrp_.rank() == 0)
         ncclGetUniqueId(&id);
-      MPI_Bcast((void*)&id, sizeof(id), MPI_BYTE, 0, &tgrp_);
+      MPI_Bcast((void*)&id, sizeof(id), MPI_BYTE, 0, tgrp_.get());
       NCCLCHECK(ncclCommInitRank(&nccl_TGcomm_, tgrp_.size(), id, tgrp_.rank()));
     }
 #endif
@@ -436,10 +436,6 @@ public:
     auto t = TGMap.find(nn);
     if (t == TGMap.end())
     {
-#if defined(ENABLE_CUDA) || defined(ENABLE_HIP)
-      if (gTG_.getTotalNodes() % nn != 0)
-        APP_ABORT("Error: nnodes must divide the total number of processors. \n\n\n");
-#endif
       auto p = TGMap.insert(
           std::make_pair(nn, afqmc::TaskGroup_(gTG_, std::string("TaskGroup_") + std::to_string(nn), nn, ncores)));
       if (!p.second)