diff --git a/srcCXX/cxx_sfx_model_compute_flux.cpp b/srcCXX/cxx_sfx_model_compute_flux.cpp
index 22213a46fb88bbf72f5a9cf8bf030ac4fc3a9f30..9367ce4c9715d5a9c4172021a5a0aa13a52073ef 100644
--- a/srcCXX/cxx_sfx_model_compute_flux.cpp
+++ b/srcCXX/cxx_sfx_model_compute_flux.cpp
@@ -31,11 +31,11 @@ void cxx_sheba_compute_flux (sfxDataVecTypeC* sfx,
                            const sfx_phys_constants* constants,
                            const int grid_size)
 {
-// #ifdef INCLUDE_CUDA
-//     static FluxSheba<float, MemType::CPU, MemType::CPU, MemType::GPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
-//     F.compute_flux();
-// #else
-//     static FluxSheba<float, MemType::CPU, MemType::CPU, MemType::CPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
-//     F.compute_flux();
-// #endif
+#ifdef INCLUDE_CUDA
+    static FluxSheba<float, MemType::CPU, MemType::CPU, MemType::GPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
+    F.compute_flux();
+#else
+    static FluxSheba<float, MemType::CPU, MemType::CPU, MemType::CPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
+    F.compute_flux();
+#endif
 }
\ No newline at end of file