diff --git a/torchtext/data/__init__.py b/torchtext/data/__init__.py
index d2aa64ccaf..b001027283 100644
--- a/torchtext/data/__init__.py
+++ b/torchtext/data/__init__.py
@@ -12,8 +12,6 @@
     to_map_style_dataset,
 )
 
-from ..legacy.data import Batch
-
 __all__ = ["bleu_score",
            "get_tokenizer", "interleave_keys",
            "generate_sp_model", "load_sp_model",
@@ -21,6 +19,4 @@
            "custom_replace", "simple_space_split",
            "numericalize_tokens_from_iterator",
            "filter_wikipedia_xml",
-           "to_map_style_dataset",
-           "Batch"  # tmp compatibility hack for old lightning
-           ]
+           "to_map_style_dataset"]