diff --git a/java-numbers/src/main/java/com/baeldung/nth/root/calculator/NthRootCalculator.java b/java-numbers/src/main/java/com/baeldung/nth/root/calculator/NthRootCalculator.java
index 217f1e06de..5e9f5bbcb4 100644
--- a/java-numbers/src/main/java/com/baeldung/nth/root/calculator/NthRootCalculator.java
+++ b/java-numbers/src/main/java/com/baeldung/nth/root/calculator/NthRootCalculator.java
@@ -1,8 +1,12 @@
 package com.baeldung.nth.root.calculator;
 
-public class NthRootCalculator
-{
-    public Double calculate(Double base, Double n) {
-        return Math.pow(Math.E, Math.log(base)/n);
+public class NthRootCalculator {
+
+    public double calculateWithRound(double base, double n) {
+        return Math.round(calculate(base, n));
+    }
+
+    public double calculate(double base, double n) {
+        return Math.pow(base, 1.0 / n);
     }
 }
diff --git a/java-numbers/src/main/java/com/baeldung/nth/root/main/Main.java b/java-numbers/src/main/java/com/baeldung/nth/root/main/Main.java
index 3fcd36812f..50190da38c 100644
--- a/java-numbers/src/main/java/com/baeldung/nth/root/main/Main.java
+++ b/java-numbers/src/main/java/com/baeldung/nth/root/main/Main.java
@@ -7,7 +7,7 @@ public class Main {
         NthRootCalculator calculator = new NthRootCalculator();
         Double base = Double.parseDouble(args[0]);
         Double n = Double.parseDouble(args[1]);
-        Double result = calculator.calculate(base, n);
+        Double result = calculator.calculateWithRound(base, n);
         System.out.println("The " + n + " root of " + base + " equals to " + result + ".");
     }
 }
diff --git a/java-numbers/src/test/java/com/baeldung/nth/root/calculator/NthRootCalculatorUnitTest.java b/java-numbers/src/test/java/com/baeldung/nth/root/calculator/NthRootCalculatorUnitTest.java
index ab962fc150..35970536ed 100644
--- a/java-numbers/src/test/java/com/baeldung/nth/root/calculator/NthRootCalculatorUnitTest.java
+++ b/java-numbers/src/test/java/com/baeldung/nth/root/calculator/NthRootCalculatorUnitTest.java
@@ -1,19 +1,22 @@
 package com.baeldung.nth.root.calculator;
 
-import static org.junit.Assert.assertEquals;
-
 import org.junit.Test;
-import org.junit.runner.RunWith;
-import org.mockito.InjectMocks;
-import org.mockito.junit.MockitoJUnitRunner;
+
+import static org.junit.Assert.assertEquals;
 
 public class NthRootCalculatorUnitTest {
 
     private NthRootCalculator nthRootCalculator = new NthRootCalculator();
 
     @Test
-    public void whenBaseIs125AndNIs3_thenNthRootIs5() {
-       Double result = nthRootCalculator.calculate(125.0, 3.0);
-       assertEquals(result, (Double) 5.0d);
+    public void whenBaseIs125AndNIs3_thenNthIs5() {
+        double nth = nthRootCalculator.calculateWithRound(125,3);
+        assertEquals(5, nth, 0);
+    }
+
+    @Test
+    public void whenBaseIs625AndNIs4_thenNthIs5() {
+        double nth = nthRootCalculator.calculate(625,4);
+        assertEquals(5, nth, 0.00001);
     }
 }
diff --git a/java-numbers/src/test/java/com/baeldung/nth/root/main/MainUnitTest.java b/java-numbers/src/test/java/com/baeldung/nth/root/main/MainUnitTest.java
deleted file mode 100644
index a2fd839ba4..0000000000
--- a/java-numbers/src/test/java/com/baeldung/nth/root/main/MainUnitTest.java
+++ /dev/null
@@ -1,34 +0,0 @@
-package com.baeldung.nth.root.main;
-
-import java.io.ByteArrayOutputStream;
-import java.io.PrintStream;
-
-import org.junit.After;
-import org.junit.Before;
-import org.junit.Test;
-import org.mockito.InjectMocks;
-import static org.junit.Assert.assertEquals;
-
-public class MainUnitTest {
-    @InjectMocks
-    private Main main;
-
-    private final ByteArrayOutputStream outContent = new ByteArrayOutputStream();
-    private final PrintStream originalOut = System.out;
-
-    @Before
-    public void setUpStreams() {
-        System.setOut(new PrintStream(outContent));
-    }
-
-    @After
-    public void restoreStreams() {
-        System.setOut(originalOut);
-    }
-
-    @Test
-    public void givenThatTheBaseIs125_andTheExpIs3_whenMainIsCalled_thenTheCorrectResultIsPrinted() {
-        main.main(new String[]{"125.0", "3.0"});
-        assertEquals("The 3.0 root of 125.0 equals to 5.0.\n", outContent.toString().replaceAll("\r", ""));
-    }
-}