From 843f55dbe2f270ffa43c7e21d679bf9b0313cf24 Mon Sep 17 00:00:00 2001
From: Kelly Davis <kdavis@mozilla.com>
Date: Thu, 20 Apr 2017 12:46:51 +0200
Subject: [PATCH] Fixed #546

---
 util/data_set_helpers.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/util/data_set_helpers.py b/util/data_set_helpers.py
index a2cf2461..8409c7fb 100644
--- a/util/data_set_helpers.py
+++ b/util/data_set_helpers.py
@@ -66,6 +66,7 @@ class DataSet(object):
         self._coord = coord
         batch_threads = [Thread(target=self._populate_batch_queue, args=(session,)) for i in range(self._thread_count)]
         for batch_thread in batch_threads:
+            self._coord.register_thread(batch_thread)
             batch_thread.daemon = True
             batch_thread.start()
         return batch_threads