1 files changed, 10 insertions, 0 deletions
diff --git a/src/transformations/xyzgridshift.cpp b/src/transformations/xyzgridshift.cpp
index f23c2588..734ffc5d 100644
--- a/src/transformations/xyzgridshift.cpp
+++ b/src/transformations/xyzgridshift.cpp
@@ -270,10 +270,20 @@ static PJ *destructor (PJ *P, int errlev) {
     return pj_default_destructor(P, errlev);
 }
 
+static void reassign_context( PJ* P, PJ_CONTEXT* ctx )
+{
+    auto Q = (struct xyzgridshiftData *) P->opaque;
+    for( auto& grid: Q->grids ) {
+        grid->reassign_context(ctx);
+    }
+}
+
+
 PJ *TRANSFORMATION(xyzgridshift,0) {
     auto Q = new xyzgridshiftData;
     P->opaque = (void *) Q;
     P->destructor = destructor;
+    P->reassign_context = reassign_context;
 
     P->fwd4d  = nullptr;
     P->inv4d  = nullptr;