diff --git a/tests/test_main.py b/tests/test_main.py
index efe816e..4c3e8e9 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -110,13 +110,14 @@ def test_flag_cmd_cli():
         assert result_std.exit_code == 0
         assert "Last detected flag: flag{test_cli_flag}" in result_std.output
         
-        # Test plain flag command output (no flavor text, no newline)
+        # Test plain flag command output (no flavor text, has newline)
         result_plain = runner.invoke(cli, ["flag", "--plain"])
         assert result_plain.exit_code == 0
-        assert result_plain.output == "flag{test_cli_flag}"
+        assert result_plain.output.strip() == "flag{test_cli_flag}"
     finally:
         write_config(original_config, str(config_file))
 
+
 def test_flag_cmd_exemption(capsys):
     """
     Verifies that running ctf flag is exempted from triggering the stdout flag warning box.