diff --git a/MLX_VERSION b/MLX_VERSION
index e115ff2bd..36bc0faa3 100644
--- a/MLX_VERSION
+++ b/MLX_VERSION
@@ -1 +1 @@
-64ea011cb65f14d9ce2737e60db9a4ae91ed7441
\ No newline at end of file
+264c14fe650e1dd9bb0c3902dc06b08033d8f89c
\ No newline at end of file
diff --git a/mlx/thread_test.go b/mlx/thread_test.go
index 7fcc29677..22069e11c 100644
--- a/mlx/thread_test.go
+++ b/mlx/thread_test.go
@@ -34,7 +34,11 @@ func mlxTestThread(tb testing.TB) *mlxthreadtest.Thread {
 
 func withMLXThread(t *testing.T, fn func(*mlxthreadtest.T)) {
 	t.Helper()
-	mlxthreadtest.Run(t, mlxTestThread(t), fn)
+	mlxthreadtest.Run(t, mlxTestThread(t), func(t *mlxthreadtest.T) {
+		Scoped(func() {
+			fn(t)
+		})
+	})
 }
 
 func TestThreadedMLXOperations(t *testing.T) {