diff --git a/testing-modules/mockito-simple/src/test/java/com/baeldung/mockito/MockFinalsUnitTest.java b/testing-modules/mockito-simple/src/test/java/com/baeldung/mockito/MockFinalsUnitTest.java
index bb03fca06f..24ab67049f 100644
--- a/testing-modules/mockito-simple/src/test/java/com/baeldung/mockito/MockFinalsUnitTest.java
+++ b/testing-modules/mockito-simple/src/test/java/com/baeldung/mockito/MockFinalsUnitTest.java
@@ -1,37 +1,32 @@
 package com.baeldung.mockito;
 
-import org.junit.Test;
-
-import static org.junit.Assert.assertNotEquals;
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
-import com.baeldung.mockito.MyList;
+import org.junit.Test;
 
 public class MockFinalsUnitTest {
 
-      @Test
-      public void whenMockFinalClassMockWorks() {
+    @Test
+    public void whenMockFinalMethodMockWorks() {
+
+        MyList myList = new MyList();
+
+        MyList mock = mock(MyList.class);
+        when(mock.finalMethod()).thenReturn(1);
+
+        assertThat(mock.finalMethod()).isNotEqualTo(myList.finalMethod());
+    }
+
+    @Test
+    public void whenMockFinalClassMockWorks() {
 
         FinalList finalList = new FinalList();
 
         FinalList mock = mock(FinalList.class);
         when(mock.size()).thenReturn(2);
 
-        assertNotEquals(mock.size(), finalList.size());
-
-      }
-
-    @Test
-    public void whenMockFinalMethodMockWorks() {
-
-      MyList myList = new MyList();
-
-      MyList mock = mock(MyList.class);
-      when(mock.finalMethod()).thenReturn(1);
-
-      assertNotEquals(mock.finalMethod(), myList.finalMethod());
+        assertThat(mock.size()).isNotEqualTo(finalList.size());
     }
-
-
-  }
+}