diff --git a/federated_language/framework/BUILD b/federated_language/framework/BUILD
index e17534e..daeb155 100644
--- a/federated_language/framework/BUILD
+++ b/federated_language/framework/BUILD
@@ -43,7 +43,6 @@ py_library(
         "//federated_language/computation:polymorphic_computation",
         "//federated_language/context_stack:context_base",
         "//federated_language/context_stack:context_stack_base",
-        "//federated_language/context_stack:context_stack_impl",
         "//federated_language/context_stack:context_stack_test_utils",
         "//federated_language/context_stack:get_context_stack",
         "//federated_language/context_stack:runtime_error_context",
diff --git a/federated_language/framework/__init__.py b/federated_language/framework/__init__.py
index 929876d..91b336d 100644
--- a/federated_language/framework/__init__.py
+++ b/federated_language/framework/__init__.py
@@ -120,7 +120,6 @@
 from federated_language.context_stack.context_base import ContextError
 from federated_language.context_stack.context_base import SyncContext
 from federated_language.context_stack.context_stack_base import ContextStack
-from federated_language.context_stack.context_stack_impl import context_stack as global_context_stack
 from federated_language.context_stack.context_stack_test_utils import with_context
 from federated_language.context_stack.context_stack_test_utils import with_contexts
 from federated_language.context_stack.get_context_stack import get_context_stack