diff --git a/obl_config.f90 b/obl_config.f90
index 12c9a46e5e2f3e8f24024ea9919cfe61041d59f3..b5e2808ed9bfa1c3d73c4466b87aba9c535430e6 100644
--- a/obl_config.f90
+++ b/obl_config.f90
@@ -370,6 +370,8 @@ contains
 
 #ifdef USE_CONFIG_PARSER
         block 
+            !> just a temporaty to set smth
+            call set_initial_conditions_kato(grid)
 
         end block
 #endif
diff --git a/obl_k_epsilon.f90 b/obl_k_epsilon.f90
index 26f54b6ff013c5c1f76665c5731a2af2c7a9c096..78b2874c8ee923515a93e1bed625f3855bb5e5db 100644
--- a/obl_k_epsilon.f90
+++ b/obl_k_epsilon.f90
@@ -103,7 +103,7 @@ module obl_k_epsilon
         real, intent(out) :: TKE(nz) !< TKE, [J/kg]
         integer :: k !< counter
 
-        do k = 2, nz-1
+        do k = 1, nz
             TKE(k) = 1.0 / (param%C_mu)**(1.0/2.0) * 0.001 * 0.001
         end do     
     end subroutine
@@ -116,7 +116,7 @@ module obl_k_epsilon
         real, intent(out) :: eps(nz) !< TKE dissipation rate, [W/kg]
         integer :: k !< counter
 
-        do k = 2, nz-1
+        do k = 1, nz
             eps(k) = 0.001 * 0.001 * 0.001 / (param%kappa * z) !not full z, but current z - must be fixed
         end do     
     end subroutine
diff --git a/obl_main.f90 b/obl_main.f90
index b88073dad87b02df460dca2680d541fc606444eb..093076022a406d31ce175c771291d77343fbcb31 100644
--- a/obl_main.f90
+++ b/obl_main.f90
@@ -243,7 +243,11 @@ program obl_main
     ! ----------------------------------------------------------------------------  
     if (closure_mode.eq.3 .or. closure_mode.eq.4) then
         call TKE_init(TKE, param_k_epsilon, grid%cz)
-        call eps_init(EPS, param_k_epsilon, grid%cz, grid%height)
+        call EPS_init(EPS, param_k_epsilon, grid%cz, grid%height)
+
+        !< have to define Km, Kh at init
+        Km = 0.0
+        Kh = 0.0
     endif
     ! ---------------------------------------------------------------------------- 
 
diff --git a/obl_run_kato.f90 b/obl_run_kato.f90
index 8b441f212df543480e7388506f5e40d426cda319..650aaa21dd2560d5e49873c471ffd5c8282c30e5 100644
--- a/obl_run_kato.f90
+++ b/obl_run_kato.f90
@@ -25,7 +25,7 @@ module obl_run_kato
     ! --------------------------------------------------------------------------------
     ! --------------------------------------------------------------------------------
     
-    
+
     contains
 
     ! --------------------------------------------------------------------------------