diff --git a/analysis/slices.py b/analysis/slices.py
index b3fa8e1..41f8fcc 100644
--- a/analysis/slices.py
+++ b/analysis/slices.py
@@ -34,7 +34,7 @@ def plot_imshow_with_reference( data_list,
     elif isinstance(L, int) or isinstance(L, float):
         L = [L for data in data_list]
 
-    sep = 10 if L[0] < 50 else 20 if L[0] < 200 else 100
+    sep = 10 if L[0] < 50 else 20 if L[0] < 100 else 50 if L[0]<250 else 100 if L[0] < 500 else 200 if L[0] < 1000 else 500 if L[0] < 2500 else 1000
     ticks = [np.arange(0, l+1, sep)*len(dat)/l for l, dat in zip(L,data_list)]
     tick_labels = [np.arange(0, l+1, sep) for l in L]
 
@@ -126,12 +126,12 @@ if __name__ == "__main__":
     parser.add_argument('-t', '--title', type=str, default=None, help='Title for the plot.')
     parser.add_argument('-log','--log_scale', action='store_true', help='Use log scale for the data.')
 
-    register_arguments_cosmo(parser)
+    # register_arguments_cosmo(parser)
 
     args = parser.parse_args()
 
     from pysbmy.field import read_field
-    from pysbmy.cosmology import d_plus
+    # from pysbmy.cosmology import d_plus
 
     ref_field = read_field(args.directory+args.reference) if args.reference is not None else None
     fields = [read_field(args.directory+f) for f in args.filenames]
@@ -173,4 +173,4 @@ if __name__ == "__main__":
     if args.output is not None:
         fig.savefig(args.output)
     else:
-        fig.savefig(args.directory+'slices.png')
\ No newline at end of file
+        fig.savefig(args.directory+'slices.png')
diff --git a/slurm_submission.py b/slurm_submission.py
index 22cea6a..78667ef 100644
--- a/slurm_submission.py
+++ b/slurm_submission.py
@@ -140,7 +140,7 @@ def create_slurm_script(slurm_template:str,
         raise ValueError(f"Array job range provided for job type {job}.")
     if array is None and job == "scola":
         raise ValueError(f"Array job range not provided for job type {job}.")
-    else:
+    elif job == "scola":
         index_sub=array[0]//limit_slurm_arrays
         array=(array[0]%limit_slurm_arrays, array[1]%limit_slurm_arrays)