diff --git a/Production/noisy_simulations_array_job.py b/Production/noisy_simulations_array_job.py
index 15f84fdf..5c73e920 100644
--- a/Production/noisy_simulations_array_job.py
+++ b/Production/noisy_simulations_array_job.py
@@ -15,9 +15,10 @@
         for num_res in num_res_arr:
             for shots in shots_arr:
                 for alpha in alpha_arr:
-                    if count == match:
-                        NSP.noisy_simulation(num_rot, num_res, shots, alpha, p_arr)
-                        break
-                    count += 1
+                    for p in p_arr:
+                        if count == match:
+                            NSP.noisy_simulation(num_rot, num_res, shots, alpha, p)
+                            break
+                        count += 1
 
 
diff --git a/Production/noisy_simulations_production.py b/Production/noisy_simulations_production.py
index 92130c09..5a997276 100644
--- a/Production/noisy_simulations_production.py
+++ b/Production/noisy_simulations_production.py
@@ -153,10 +153,7 @@ def callback(quasi_dists, parameters, energy):
 
 
 
-    # %%
-    import pandas as pd
-    import numpy as np
-    from matplotlib import pyplot as plt
+