diff --git a/router/java/src/net/i2p/router/peermanager/PeerTestJob.java b/router/java/src/net/i2p/router/peermanager/PeerTestJob.java index 903386189e29c87f191ec2a8bd168a1cc53f95c8..ac1918808291c4cb3a387d6d59be498dd414210d 100644 --- a/router/java/src/net/i2p/router/peermanager/PeerTestJob.java +++ b/router/java/src/net/i2p/router/peermanager/PeerTestJob.java @@ -207,7 +207,7 @@ public class PeerTestJob extends JobImpl { public boolean continueMatching() { return false; } public long getExpiration() { return _expiration; } public boolean isMatch(I2NPMessage message) { - if (message instanceof DeliveryStatusMessage) { + if (message.getType() == DeliveryStatusMessage.MESSAGE_TYPE) { DeliveryStatusMessage msg = (DeliveryStatusMessage)message; if (_nonce == msg.getMessageId()) { long timeLeft = _expiration - getContext().clock().now(); diff --git a/router/java/src/net/i2p/router/tunnel/pool/TestJob.java b/router/java/src/net/i2p/router/tunnel/pool/TestJob.java index 52923fa720a513ff41694a0d8a63f8e64337ef56..1bbb48fba61fb77fc46b57b4b741ab7aec2a0540 100644 --- a/router/java/src/net/i2p/router/tunnel/pool/TestJob.java +++ b/router/java/src/net/i2p/router/tunnel/pool/TestJob.java @@ -267,7 +267,7 @@ class TestJob extends JobImpl { public long getExpiration() { return _expiration; } public boolean isMatch(I2NPMessage message) { - if (message instanceof DeliveryStatusMessage) { + if (message.getType() == DeliveryStatusMessage.MESSAGE_TYPE) { return ((DeliveryStatusMessage)message).getMessageId() == _id; } return false;