diff --git a/undertow/src/main/java/com/baeldung/undertow/secure/CustomIdentityManager.java b/undertow/src/main/java/com/baeldung/undertow/secure/CustomIdentityManager.java
index e0984f65a5..16231f036d 100644
--- a/undertow/src/main/java/com/baeldung/undertow/secure/CustomIdentityManager.java
+++ b/undertow/src/main/java/com/baeldung/undertow/secure/CustomIdentityManager.java
@@ -8,9 +8,10 @@ import java.util.Set;
 
 import io.undertow.security.idm.Account;
 import io.undertow.security.idm.Credential;
+import io.undertow.security.idm.IdentityManager;
 import io.undertow.security.idm.PasswordCredential;
 
-public class CustomIdentityManager implements io.undertow.security.idm.IdentityManager {
+public class CustomIdentityManager implements IdentityManager {
 
     private final Map<String, char[]> users;
 
diff --git a/undertow/src/main/java/com/baeldung/undertow/secure/SecureServer.java b/undertow/src/main/java/com/baeldung/undertow/secure/SecureServer.java
index 6f520944db..9997883da6 100644
--- a/undertow/src/main/java/com/baeldung/undertow/secure/SecureServer.java
+++ b/undertow/src/main/java/com/baeldung/undertow/secure/SecureServer.java
@@ -29,13 +29,15 @@ public class SecureServer {
         final IdentityManager idm = new CustomIdentityManager(users);
 
         Undertow server = Undertow.builder().addHttpListener(8080, "localhost").setHandler(addSecurity((exchange) -> {
-            final SecurityContext context = exchange.getSecurityContext();
-            exchange.getResponseSender().send("Hello " + context.getAuthenticatedAccount().getPrincipal().getName(),
-                    IoCallback.END_EXCHANGE);
+            setExchange(exchange);
         }, idm)).build();
 
         server.start();
+    }
 
+    private static void setExchange(HttpServerExchange exchange) {
+        final SecurityContext context = exchange.getSecurityContext();
+        exchange.getResponseSender().send("Hello " + context.getAuthenticatedAccount().getPrincipal().getName(),IoCallback.END_EXCHANGE);
     }
 
     private static HttpHandler addSecurity(final HttpHandler toWrap, final IdentityManager identityManager) {