diff --git a/extension/training/optimizer/targets.bzl b/extension/training/optimizer/targets.bzl
index 84043d27c9..69682feaee 100644
--- a/extension/training/optimizer/targets.bzl
+++ b/extension/training/optimizer/targets.bzl
@@ -26,7 +26,7 @@ def define_common_targets():
             ]
 
         runtime.cxx_library(
-            name = "optimizer" + aten_suffix,
+            name = "sgd" + aten_suffix,
             srcs = [
                 "sgd.cpp",
             ],
diff --git a/extension/training/optimizer/test/targets.bzl b/extension/training/optimizer/test/targets.bzl
index 7ffa74d614..11269bfa18 100644
--- a/extension/training/optimizer/test/targets.bzl
+++ b/extension/training/optimizer/test/targets.bzl
@@ -15,7 +15,7 @@ def define_common_targets():
                 "sgd_test.cpp",
             ],
             deps = [
-                "//executorch/extension/training/optimizer:optimizer" + aten_suffix,
+                "//executorch/extension/training/optimizer:sgd" + aten_suffix,
                 "//executorch/runtime/core:core",
                 "//executorch/runtime/core/exec_aten/testing_util:tensor_util",
             ],