diff --git a/src/test/java/ysoserial/payloads/TestHarnessTest.java b/src/test/java/ysoserial/payloads/TestHarnessTest.java index 09df5dc..33ce7f7 100755 --- a/src/test/java/ysoserial/payloads/TestHarnessTest.java +++ b/src/test/java/ysoserial/payloads/TestHarnessTest.java @@ -38,8 +38,8 @@ public class TestHarnessTest { PayloadsTest.testPayload(ExecMockPayload.class, new Class[] { ExecMockSerializable.class }); } - public static class ExecMockPayload implements ObjectPayload { - public ExecMockSerializable getObject(String command) throws Exception { + public static class ExecMockPayload extends ExtendedObjectPayload { + public ExecMockSerializable getObject(String[] command) throws Exception { return new ExecMockSerializable(command); } } @@ -52,8 +52,8 @@ public class TestHarnessTest { @SuppressWarnings("serial") public static class ExecMockSerializable implements Serializable { - private final String cmd; - public ExecMockSerializable(String cmd) { this.cmd = cmd; } + private final String[] cmd; + public ExecMockSerializable(String[] cmd) { this.cmd = cmd; } private void readObject(final ObjectInputStream ois) throws IOException, ClassNotFoundException { ois.defaultReadObject();