diff --git a/tests/benchmarks/heat-equation-benchmark/HeatEquationBenchmarkProblem_impl.h b/tests/benchmarks/heat-equation-benchmark/HeatEquationBenchmarkProblem_impl.h
index 403044b3cf187c11856a58be98e323094fe75d83..e2ba8ebcdcc23007a6ef54578ddc0ef9ab90dbf3 100644
--- a/tests/benchmarks/heat-equation-benchmark/HeatEquationBenchmarkProblem_impl.h
+++ b/tests/benchmarks/heat-equation-benchmark/HeatEquationBenchmarkProblem_impl.h
@@ -536,7 +536,6 @@ getExplicitUpdate( const RealType& time,
          cudaThreadSynchronize();         
          TNL_CHECK_CUDA_DEVICE;
       }
-      #endif
       if( this->cudaKernelType == "templated" )
       {
          //if( !this->cudaMesh )
@@ -641,6 +640,7 @@ getExplicitUpdate( const RealType& time,
             
          }
       }      
+      #endif
    }
 }