diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py index 43f98c1c1efb61..aa1c6013dcc1de 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -28,7 +28,7 @@ def load_tests(loader, tests, ignore): )) return tests -MODULE = ('test.test_enum', '__main__')[__name__=='__main__'] +MODULE = __name__ SHORT_MODULE = MODULE.split('.')[-1] # for pickle tests