diff --git a/featuretools/computational_backends/calculate_feature_matrix.py b/featuretools/computational_backends/calculate_feature_matrix.py index d91943d37e..435249b787 100644 --- a/featuretools/computational_backends/calculate_feature_matrix.py +++ b/featuretools/computational_backends/calculate_feature_matrix.py @@ -744,7 +744,7 @@ def parallel_calculate_chunks( _saved_features = client.scatter(pickled_feats) client.replicate([_es, _saved_features]) num_scattered_workers = len( - client.who_has([Future(es_token)]).get(es_token, []), + client.who_has([Future(es_token, client=client)]).get(es_token, []), ) num_workers = len(client.scheduler_info()["workers"].values())