diff --git a/python/tffpy/scripts/script_exp_solve_tff.py b/python/tffpy/scripts/script_exp_solve_tff.py
index 3cebe337de35065b3dfa2eae7db78cfde6ceefd8..6ad389d0617c0ec7c5f6c1ee7ce78b1b3c0a895d 100644
--- a/python/tffpy/scripts/script_exp_solve_tff.py
+++ b/python/tffpy/scripts/script_exp_solve_tff.py
@@ -52,12 +52,11 @@ Run this script to handle the main experiment :class:`SolveTffExperiment`.
 
 .. moduleauthor:: Valentin Emiya
 """
-from yafe.utils import generate_oar_script
-
 import matplotlib.pyplot as plt
 
 from tffpy.experiments.exp_solve_tff import \
     SolveTffExperiment, create_and_run_light_experiment
+from tffpy.experiments.yafe_slurm import generate_slurm_script
 
 
 try:
@@ -77,7 +76,7 @@ if __name__ == '__main__':
         input_msg = '\n'.join(['1 - Create and run light experiment',
                                '2 - Display results of light experiment',
                                '3 - Full experiment: create full experiment',
-                               '4 - Generate OAR script',
+                               '4 - Generate Slurm script',
                                '5 - Full experiment: collect results',
                                '6 - Full experiment: download results',
                                '7 - Full experiment: display results',
@@ -104,12 +103,12 @@ if __name__ == '__main__':
         elif answer == 4:
             experiment.display_status()
             batch_size = int(input('Batch size (#tasks per job)?'))
-            generate_oar_script(script_file_path=__file__,
-                                xp_var_name='experiment',
-                                batch_size=batch_size,
-                                oar_walltime='01:00:00',
-                                activate_env_command='source activate py36',
-                                use_gpu=False)
+            generate_slurm_script(script_file_path=__file__,
+                                  xp_var_name='experiment',
+                                  n_simultaneous_jobs=n_simultaneous_jobs,
+                                  slurm_walltime='02:00:00',
+                                  activate_env_command='source activate py36',
+                                  use_cpu_gpu='cpu')
         elif answer == 5:
             experiment.collect_results()
             experiment.display_status()
diff --git a/python/tffpy/scripts/script_exp_variance.py b/python/tffpy/scripts/script_exp_variance.py
index 7155cdbebd0dfa73a3e9c92fc9d857ad0d55d745..5ddb2b4de600368608e3c3985af83deb6c47745f 100644
--- a/python/tffpy/scripts/script_exp_variance.py
+++ b/python/tffpy/scripts/script_exp_variance.py
@@ -104,7 +104,7 @@ if __name__ == '__main__':
                                   n_simultaneous_jobs=n_simultaneous_jobs,
                                   slurm_walltime='02:00:00',
                                   activate_env_command='source activate py36',
-                                  use_gpu=False)
+                                  use_cpu_gpu='cpu')
         elif answer == 5:
             experiment.collect_results()
             experiment.display_status()