diff --git a/2.11-Exercise-WorkWithUnitTests/src/main/java/com/udacity/gradle/Echo.java b/2.11-Exercise-WorkWithUnitTests/src/main/java/com/udacity/gradle/Echo.java
index a221e65f8..74eee1a96 100644
--- a/2.11-Exercise-WorkWithUnitTests/src/main/java/com/udacity/gradle/Echo.java
+++ b/2.11-Exercise-WorkWithUnitTests/src/main/java/com/udacity/gradle/Echo.java
@@ -4,4 +4,4 @@ public abstract class Echo {
     public static Object echo(Object input) {
         return input;
     }
-}
\ No newline at end of file
+}
diff --git a/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/EchoTest.java b/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/EchoTest.java
index 5e9edb463..7c4ad53b8 100644
--- a/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/EchoTest.java
+++ b/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/EchoTest.java
@@ -1,11 +1,12 @@
 package com.udacity.gradle.test;
 
 import com.udacity.gradle.Echo;
+import org.junit.Assert;
 import org.junit.Test;
 
 public class EchoTest {
     @Test
     public void test() {
-        assert Echo.echo("hello").equals("hello");
+        Assert.assertEquals(Echo.echo("hello"), "hello");
     }
-}
\ No newline at end of file
+}
diff --git a/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/PersonTest.java b/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/PersonTest.java
index 3cf322907..f40188665 100644
--- a/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/PersonTest.java
+++ b/2.11-Exercise-WorkWithUnitTests/src/test/java/com/udacity/gradle/test/PersonTest.java
@@ -1,12 +1,13 @@
 package com.udacity.gradle.test;
 
 import com.udacity.gradle.Person;
+import org.junit.Assert;
 import org.junit.Test;
 
 public class PersonTest {
     @Test
     public void test() {
         Person person = new Person("Jeremy");
-        assert person.getName().equals("Jeremy");
+        Assert.assertEquals(person.getName(), "Jeremy");
     }
-}
\ No newline at end of file
+}