diff --git a/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py b/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py
index 30312d68..ea0b1673 100644
--- a/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py
+++ b/GPy/kern/src/psi_comp/rbf_psi_gpucomp.py
@@ -360,7 +360,10 @@ class PSICOMP_RBF_GPU(PSICOMP_RBF):
         if self.GPU_direct:
             return psi0, psi1_gpu, psi2_gpu
         else:
-            return psi0, psi1_gpu.get(), psi2_gpu.get()
+            if return_psi2_n:
+                return psi0, psi1_gpu.get(), psi2n_gpu.get()
+            else:
+                return psi0, psi1_gpu.get(), psi2_gpu.get()
         
     def psiDerivativecomputations(self, kern, dL_dpsi0, dL_dpsi1, dL_dpsi2, Z, variational_posterior):
         try: