diff --git a/test/run_test.py b/test/run_test.py
index e90a991161c..370d96d00cb 100755
--- a/test/run_test.py
+++ b/test/run_test.py
@@ -1289,7 +1289,7 @@ def can_run_in_pytest(test):
     return os.getenv("PYTORCH_TEST_DO_NOT_USE_PYTEST", "0") == "0"
 
 
-def get_selected_tests(options) -> List[ShardedTest]:
+def get_selected_tests(options) -> List[str]:
     selected_tests = options.include
 
     # filter if there's JIT only and distributed only test options