diff --git a/kernel_tuner/hyper.py b/kernel_tuner/hyper.py
index 0aacca6c..b94c5898 100644
--- a/kernel_tuner/hyper.py
+++ b/kernel_tuner/hyper.py
@@ -86,19 +86,12 @@ def put_if_not_present(target_dict, key, value):
     return list(result_unique.values()), env
 
 if __name__ == "__main__":  # TODO remove in production
-    # hyperparams = {
-    #     'popsize': [10, 20, 30],
-    #     'maxiter': [50, 100, 150],
-    #     'w': [0.25, 0.5, 0.75],
-    #     'c1': [1.0, 2.0, 3.0],
-    #     'c2': [0.5, 1.0, 1.5]
-    # }
     hyperparams = {
-        'popsize': [10],
-        'maxiter': [50],
-        'w': [0.25, 0.5],
-        'c1': [1.0],
-        'c2': [0.5]
+        'popsize': [10, 20, 30],
+        'maxiter': [50, 100, 150],
+        'w': [0.25, 0.5, 0.75],
+        'c1': [1.0, 2.0, 3.0],
+        'c2': [0.5, 1.0, 1.5]
     }
     result, env = tune_hyper_params('pso', hyperparams)
     print(result)