diff --git a/mlx/compile.cpp b/mlx/compile.cpp
index 91660effd..2e671f6f8 100644
--- a/mlx/compile.cpp
+++ b/mlx/compile.cpp
@@ -1,5 +1,4 @@
 // Copyright © 2023 Apple Inc.
-#include <iostream> // TODO
 
 #include <map>
 #include <unordered_map>
diff --git a/python/src/transforms.cpp b/python/src/transforms.cpp
index 043dd8ea5..ab7bbc385 100644
--- a/python/src/transforms.cpp
+++ b/python/src/transforms.cpp
@@ -1,5 +1,4 @@
 // Copyright © 2023 Apple Inc.
-#include <iostream> // TODO
 
 #include <pybind11/functional.h>
 #include <pybind11/pybind11.h>