diff --git a/mlx/io/threadpool.h b/mlx/io/threadpool.h
index cdb3f7faf..0ed985948 100644
--- a/mlx/io/threadpool.h
+++ b/mlx/io/threadpool.h
@@ -37,7 +37,7 @@ class ThreadPool {
   ThreadPool(size_t);
   template <class F, class... Args>
   auto enqueue(F&& f, Args&&... args)
-      -> std::future<typename std::result_of_t<F(Args...)>>;
+      -> std::future<typename std::invoke_result_t<F, Args...>>;
   ~ThreadPool();
 
  private:
@@ -71,8 +71,8 @@ inline ThreadPool::ThreadPool(size_t threads) : stop(false) {
 
 template <class F, class... Args>
 auto ThreadPool::enqueue(F&& f, Args&&... args)
-    -> std::future<typename std::result_of_t<F(Args...)>> {
-  using return_type = typename std::result_of_t<F(Args...)>;
+    -> std::future<typename std::invoke_result_t<F, Args...>> {
+  using return_type = typename std::invoke_result_t<F, Args...>;
 
   auto task = std::make_shared<std::packaged_task<return_type()>>(
       std::bind(std::forward<F>(f), std::forward<Args>(args)...));