diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
index d3bf55373..6b4319843 100644
--- a/test/smoke_test/smoke_test.py
+++ b/test/smoke_test/smoke_test.py
@@ -31,6 +31,4 @@ def stateful_dataloader_test():
     if options.s3:
         s3_test()
 
-    if torchdata.__version__ != "0.8.0":
-        raise Exception("TorchData version is not 0.8.0")
     stateful_dataloader_test()