From 72de587568bf8e54979a38878c8b937c0cd094fa Mon Sep 17 00:00:00 2001
From: Chris Frohoff <chris@frohoff.org>
Date: Sun, 5 Jun 2022 23:36:44 -0700
Subject: [PATCH] fix payload name parsing

---
 src/test/java/ysoserial/test/util/PayloadListener.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test/java/ysoserial/test/util/PayloadListener.java b/src/test/java/ysoserial/test/util/PayloadListener.java
index 59a64a8..1d2053e 100644
--- a/src/test/java/ysoserial/test/util/PayloadListener.java
+++ b/src/test/java/ysoserial/test/util/PayloadListener.java
@@ -90,6 +90,6 @@ public class PayloadListener extends RunListener {
 
     // testPayload[payloadClass: class ysoserial.payloads.JavassistWeld1](ysoserial.test.payloads.PayloadsTest)
     public static String getPayload(String displayName) {
-        return displayName.replaceAll(".*\\[\\S+: class (\\w+\\.)+(\\w+)\\].*", "$2");
+        return displayName.replaceAll(".*\\[\\S+: class (\\w+[.$])+(\\w+)\\].*", "$2");
     }
 }