From 7c8f79b1479e9f15d1c82c27579f8741757a41f8 Mon Sep 17 00:00:00 2001
From: Antonin Portelli <antonin.portelli@me.com>
Date: Fri, 21 Oct 2016 15:20:12 +0100
Subject: [PATCH] more stochastic QED fixes

---
 lib/qcd/action/Actions.h      | 1 +
 lib/qcd/action/gauge/Photon.h | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/qcd/action/Actions.h b/lib/qcd/action/Actions.h
index ba6e577d..a23c6495 100644
--- a/lib/qcd/action/Actions.h
+++ b/lib/qcd/action/Actions.h
@@ -57,6 +57,7 @@ Author: paboyle <paboyle@ph.ed.ac.uk>
 ////////////////////////////////////////////
 // Gauge Actions
 ////////////////////////////////////////////
+#include <Grid/qcd/action/gauge/Photon.h>
 #include <Grid/qcd/action/gauge/WilsonGaugeAction.h>
 #include <Grid/qcd/action/gauge/PlaqPlusRectangleAction.h>
 
diff --git a/lib/qcd/action/gauge/Photon.h b/lib/qcd/action/gauge/Photon.h
index aac72898..0dd5a9dc 100644
--- a/lib/qcd/action/gauge/Photon.h
+++ b/lib/qcd/action/gauge/Photon.h
@@ -124,7 +124,7 @@ namespace QCD{
           LatticeCoordinate(coor,d);
           spNrm = spNrm + coor*coor;
         }
-        out = where(spNrm == 0, 0.*out, out);
+        out = where(spNrm == Integer(0), 0.*out, out);
         
         break;
       }
@@ -149,7 +149,7 @@ namespace QCD{
   template<class Gimpl>
   void Photon<Gimpl>::StochasticField(GaugeField &out, GridParallelRNG &rng)
   {
-    auto               *grid = out._grid;
+    auto               *grid = dynamic_cast<GridCartesian *>(out._grid);
     const unsigned int nd = grid->_ndimension;
     GaugeLinkField     sqrtK2Inv(grid), r(grid);
     GaugeField         aTilde(grid);