diff --git a/mlx/ops.h b/mlx/ops.h
index d8bc05c72..c0feaa402 100644
--- a/mlx/ops.h
+++ b/mlx/ops.h
@@ -1057,7 +1057,7 @@ array dequantize(
     int bits = 4,
     StreamOrDevice s = {});
 
-/** Load array from .safetensor file format */
+/** Load array map from .safetensor file format */
 std::unordered_map<std::string, array> load_safetensor(
     std::shared_ptr<io::Reader> in_stream,
     StreamOrDevice s = {});
diff --git a/mlx/safetensor.cpp b/mlx/safetensor.cpp
index 1ec3f2964..37d7ceb22 100644
--- a/mlx/safetensor.cpp
+++ b/mlx/safetensor.cpp
@@ -85,7 +85,6 @@ std::unordered_map<std::string, array> load_safetensor(
   char rawJson[jsonHeaderLength];
   in_stream->read(rawJson, jsonHeaderLength);
   auto metadata = json::parse(rawJson, rawJson + jsonHeaderLength);
-  // auto metadata = io::jsonDeserialize(json, jsonHeaderLength);
   // Should always be an object on the top-level
   if (!metadata.is_object()) {
     throw std::runtime_error(