From 42a80cdaf689b03b326164862318039bd43bbff1 Mon Sep 17 00:00:00 2001
From: Nivid Patel <66813410+nivid26@users.noreply.github.com>
Date: Wed, 22 Jun 2022 00:04:18 -0400
Subject: [PATCH] Update basic_maths.py (#6017)

---
 maths/basic_maths.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/maths/basic_maths.py b/maths/basic_maths.py
index 58e797772..26c52c549 100644
--- a/maths/basic_maths.py
+++ b/maths/basic_maths.py
@@ -57,6 +57,8 @@ def number_of_divisors(n: int) -> int:
             temp += 1
             n = int(n / i)
         div *= temp
+    if n > 1:
+        div *= 2
     return div