diff --git a/tests/integration-tests/tests/ad_integration/test_ad_integration.py b/tests/integration-tests/tests/ad_integration/test_ad_integration.py
index 5a46cbb62f..6ea296c079 100644
--- a/tests/integration-tests/tests/ad_integration/test_ad_integration.py
+++ b/tests/integration-tests/tests/ad_integration/test_ad_integration.py
@@ -449,10 +449,9 @@ def _check_whoami(users):
     logging.info("Checking whoami")
     for user in users:
         result = user.run_remote_command("whoami").stdout
-        # TODO: Change the assertion to check the output is just the Username
-        assert_that(result).contains(user.alias)
+        assert_that(result).is_equal_to(user.alias)
         result = user.run_remote_command("srun whoami").stdout
-        assert_that(result).contains(user.alias)
+        assert_that(result).is_equal_to(user.alias)
 
 
 def _check_files_permissions(users, shared_storage_mount_dirs):