diff --git a/testing/test_doctest.py b/testing/test_doctest.py
index 67b8ccdb7..1b4809240 100644
--- a/testing/test_doctest.py
+++ b/testing/test_doctest.py
@@ -803,8 +803,8 @@ class TestDoctests:
         """
         p = pytester.makepyfile(
             setup="""
-            from setuptools import setup, find_packages
             if __name__ == '__main__':
+                from setuptools import setup, find_packages
                 setup(name='sample',
                       version='0.0',
                       description='description',