diff --git a/src/lbm/collision/TurbulentViscosity.h b/src/lbm/collision/TurbulentViscosity.h
index 0122dda7710f0832511afb83103c97ebcda9a384..8ec8130b79b7e3105fb4817a139db2a686627276 100644
--- a/src/lbm/collision/TurbulentViscosity.h
+++ b/src/lbm/collision/TurbulentViscosity.h
@@ -71,7 +71,7 @@ inline __host__ __device__ real calcTurbulentViscositySmagorinsky(real Cs, real
 }
 
 template <typename T>
-__host__ __device__ int max( T a, T b )
+__host__ __device__ T max( T a, T b )
 {
     return ( a > b ) ? a : b;
 }
@@ -87,6 +87,11 @@ inline __host__ __device__ real calcTurbulentViscosityQR(real C, real dxux, real
     return C * max(R, c0o1) / Q;
 }
 
+inline __host__ __device__ real calculateOmegaWithturbulentViscosity(const real omega_old, real turbulenceViscosity)
+{
+    return omega_old / (c1o1 + c3o1 * omega_old * turbulenceViscosity);
+}
+
 } // namespace vf::lbm
 
 #endif //TURBULENT_VISCOSITY_H