diff --git a/fbgemm_gpu/test/sparse/common.py b/fbgemm_gpu/test/sparse/common.py
index 8abddca752..02729f87a7 100644
--- a/fbgemm_gpu/test/sparse/common.py
+++ b/fbgemm_gpu/test/sparse/common.py
@@ -16,7 +16,7 @@
 import fbgemm_gpu
 import torch
 from hypothesis import HealthCheck, settings
-from torch._utils_internal import get_file_path_2  # @manual=//caffe2:utils_internal
+from torch._utils_internal import get_file_path_2  # @manual=//caffe2:_utils_internal
 from torch.testing._internal.optests import generate_opcheck_tests
 
 # pyre-fixme[16]: Module `fbgemm_gpu` has no attribute `open_source`.
diff --git a/fbgemm_gpu/test/test_utils.py b/fbgemm_gpu/test/test_utils.py
index 522c822e57..b7e7a8e306 100644
--- a/fbgemm_gpu/test/test_utils.py
+++ b/fbgemm_gpu/test/test_utils.py
@@ -141,7 +141,7 @@ def decorator(test_class: unittest.TestCase) -> unittest.TestCase:
             if not has_optests():
                 return test_class
             import torch.testing._internal.optests as optests
-            from torch._utils_internal import (  # @manual=//caffe2:utils_internal
+            from torch._utils_internal import (  # @manual=//caffe2:_utils_internal
                 get_file_path_2,
             )