diff --git a/core-java-9/pom.xml b/core-java-9/pom.xml
index 23473ff161..f6792fdbd9 100644
--- a/core-java-9/pom.xml
+++ b/core-java-9/pom.xml
@@ -47,6 +47,13 @@
             <version>${mockito.version}</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>com.jayway.awaitility</groupId>
+            <artifactId>awaitility</artifactId>
+            <version>${awaitility.version}</version>
+            <scope>test</scope>
+        </dependency>
+
 
     </dependencies>
 
@@ -90,6 +97,7 @@
         <org.hamcrest.version>1.3</org.hamcrest.version>
         <junit.version>4.12</junit.version>
         <mockito.version>1.10.19</mockito.version>
+        <awaitility.version>1.7.0</awaitility.version>
     </properties>
 
 </project>
diff --git a/core-java-9/src/test/java/com/baeldung/java9/streams.reactive/ReactiveStreamsTest.java b/core-java-9/src/test/java/com/baeldung/java9/streams.reactive/ReactiveStreamsTest.java
index b35d2b78eb..647557532d 100644
--- a/core-java-9/src/test/java/com/baeldung/java9/streams.reactive/ReactiveStreamsTest.java
+++ b/core-java-9/src/test/java/com/baeldung/java9/streams.reactive/ReactiveStreamsTest.java
@@ -24,8 +24,10 @@ public class ReactiveStreamsTest {
         publisher.close();
 
         //then
-        Thread.sleep(1000);
-        assertThat(subscriber.consumedElements).containsExactlyElementsOf(items);
+
+        await().atMost(1000, TimeUnit.MILLISECONDS).until(
+                () -> assertThat(subscriber.consumedElements).containsExactlyElementsOf(items)
+        );
     }
 
     @Test
@@ -44,8 +46,9 @@ public class ReactiveStreamsTest {
         publisher.close();
 
         //then
-        Thread.sleep(1000);
-        assertThat(subscriber.consumedElements).containsExactlyElementsOf(expectedResult);
+        await().atMost(1000, TimeUnit.MILLISECONDS).until(
+                () -> assertThat(subscriber.consumedElements).containsExactlyElementsOf(expectedResult)
+        );
     }
 
     @Test
@@ -63,7 +66,8 @@ public class ReactiveStreamsTest {
         publisher.close();
 
         //then
-        Thread.sleep(1000);
-        assertThat(subscriber.consumedElements).containsExactlyElementsOf(expected);
+        await().atMost(1000, TimeUnit.MILLISECONDS).until(
+                () -> assertThat(subscriber.consumedElements).containsExactlyElementsOf(expected)
+        );
     }
 }