diff --git a/auto_ml/predictor.py b/auto_ml/predictor.py
index <HASH>..<HASH> 100644
--- a/auto_ml/predictor.py
+++ b/auto_ml/predictor.py
@@ -914,6 +914,9 @@ class Predictor(object):
 
         n_jobs = -1
 
+        if os.environ.get('is_test_suite', 0) == 'True':
+            n_jobs = 1
+
         gs = GridSearchCV(
             # Fit on the pipeline.
             ppl,
@@ -1181,7 +1184,7 @@ class Predictor(object):
 
         if os.environ.get('is_test_suite', False) == 'True':
             # If this is the test_suite, do not run things in parallel
-            results = list(pool.map(lambda x: train_one_categorical_model(x[0], x[1], x[2]), categories_and_data))
+            results = list(map(lambda x: train_one_categorical_model(x[0], x[1], x[2]), categories_and_data))
         else:
             try:
                 results = list(pool.map(lambda x: train_one_categorical_model(x[0], x[1], x[2]), categories_and_data))
reinstates one place for n_jobs=1 for is_test_suite