diff --git a/docs/src/_static/mlx_logo.png b/docs/src/_static/mlx_logo.png
index be122bf..2e3dfca 100644
Binary files a/docs/src/_static/mlx_logo.png and b/docs/src/_static/mlx_logo.png differ
diff --git a/docs/src/_static/mlx_logo_dark.png b/docs/src/_static/mlx_logo_dark.png
index cda3c1f..0703b30 100644
Binary files a/docs/src/_static/mlx_logo_dark.png and b/docs/src/_static/mlx_logo_dark.png differ
diff --git a/mlx/data/core/ThreadPool.cpp b/mlx/data/core/ThreadPool.cpp
index 6d2c986..f70ac1a 100644
--- a/mlx/data/core/ThreadPool.cpp
+++ b/mlx/data/core/ThreadPool.cpp
@@ -51,7 +51,10 @@ ThreadPool::ThreadPool(size_t thread_count) {
 }
 
 ThreadPool::~ThreadPool() {
-  stop_threads_ = true;
+  {
+    std::lock_guard<std::mutex> lock(task_mutex_);
+    stop_threads_ = true;
+  }
   task_cv_.notify_all();
 
   for (std::thread& thread : threads_) {