From e49112ce2bf9818f3edd76e04a7a7c2e4242433d Mon Sep 17 00:00:00 2001
From: Neetika23 <42495275+Neetika23@users.noreply.github.com>
Date: Wed, 13 Dec 2023 23:20:55 +0530
Subject: [PATCH] Update Print2DArraysUnitTest.java

---
 .../com/baeldung/print2DArrays/Print2DArraysUnitTest.java | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/core-java-modules/core-java-arrays-guides/src/test/java/com/baeldung/print2DArrays/Print2DArraysUnitTest.java b/core-java-modules/core-java-arrays-guides/src/test/java/com/baeldung/print2DArrays/Print2DArraysUnitTest.java
index c89b1469ff..dcb027c715 100644
--- a/core-java-modules/core-java-arrays-guides/src/test/java/com/baeldung/print2DArrays/Print2DArraysUnitTest.java
+++ b/core-java-modules/core-java-arrays-guides/src/test/java/com/baeldung/print2DArrays/Print2DArraysUnitTest.java
@@ -5,10 +5,12 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
 
 import java.util.Arrays;
 
-import uk.org.webcompere.systemstubs.SystemOut;
-import uk.org.webcompere.systemstubs.annotations.SystemStub;
+import uk.org.webcompere.systemstubs.jupiter.SystemStub;
+import uk.org.webcompere.systemstubs.stream.SystemOut;
+import uk.org.webcompere.systemstubs.jupiter.SystemStubsExtension;
 
-public class Print2DArrayUnitTest {
+@ExtendWith(SystemStubsExtension.class)
+public class Print2DArraysUnitTest {
     @SystemStub
     private SystemOut systemOut;