diff --git a/torchao/__init__.py b/torchao/__init__.py
index 340bfe3013..c982e09a0c 100644
--- a/torchao/__init__.py
+++ b/torchao/__init__.py
@@ -5,8 +5,12 @@
 )
 from . import dtypes
 import torch
-from torch.testing._internal.common_utils import IS_FBCODE
-if not IS_FBCODE:
+_IS_FBCODE = (
+    hasattr(torch._utils_internal, "IS_FBSOURCE") and
+    torch._utils_internal.IS_FBSOURCE
+)
+
+if not _IS_FBCODE:
     from . import _C
     from . import ops