From 76536493894a6a50457d95ff2c5388e0f2ddd033 Mon Sep 17 00:00:00 2001
From: Felix Erben <dc-erbe1@tesseract-login1.ib0.sgi.cluster.dirac.ed.ac.uk>
Date: Fri, 1 Mar 2019 12:57:41 +0000
Subject: [PATCH] baryons working now

---
 Grid/qcd/utils/A2Autils.h | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/Grid/qcd/utils/A2Autils.h b/Grid/qcd/utils/A2Autils.h
index b3914f33..d5feb275 100644
--- a/Grid/qcd/utils/A2Autils.h
+++ b/Grid/qcd/utils/A2Autils.h
@@ -134,9 +134,7 @@ void A2Autils<FImpl>::NucleonFieldMom(Eigen::Tensor<ComplexD,6> &mat,
   int twoBlock = mat.dimension(3);
   int threeBlock = mat.dimension(4);
 
-  assert(0 && "Apologies, Felix, next line was causing compile failure");
-  //GridBase *grid = wi[0]._grid;
-  GridBase *grid = nullptr;
+  GridBase *grid = one[0]._grid;
 
   const int    nd = grid->_ndimension;
   const int Nsimd = grid->Nsimd();
@@ -202,9 +200,7 @@ void A2Autils<FImpl>::NucleonFieldMom(Eigen::Tensor<ComplexD,6> &mat,
 
 	    
 	      // After getting the sitewise product do the mom phase loop
-              assert(0 && "Apologies, Felix, next line was causing compile failure");
-              //int base = Nmom*i+Nmom*Lblock*j+Nmom*Lblock*Rblock*r;
-	      int base = 0;
+              int base = Nmom*i+Nmom*oneBlock*j+Nmom*oneBlock*twoBlock*k+Nmom*oneBlock*twoBlock*threeBlock*r;
 	      for ( int m=0;m<Nmom;m++){
 	        int idx = m+base;
 	        auto phase = mom[m]._odata[ss];