diff --git a/maths/average_median.py b/maths/average_median.py
index f24e525736b3..7d2445fd7cab 100644
--- a/maths/average_median.py
+++ b/maths/average_median.py
@@ -31,11 +31,7 @@ def median(nums: list) -> int | float:
     )
 
 
-def main():
+if __name__ == "__main__":
     import doctest
 
     doctest.testmod()
-
-
-if __name__ == "__main__":
-    main()