From 9886facf42ef130511bdc9e58b8ded422a3c3927 Mon Sep 17 00:00:00 2001
From: David Kocher <dkocher@sudo.ch>
Date: Wed, 18 Jun 2014 10:08:07 +0200
Subject: [PATCH] Fix test.

---
 src/test/java/net/schmizz/sshj/sftp/PacketReaderTest.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/test/java/net/schmizz/sshj/sftp/PacketReaderTest.java b/src/test/java/net/schmizz/sshj/sftp/PacketReaderTest.java
index b638c795..5a380be5 100644
--- a/src/test/java/net/schmizz/sshj/sftp/PacketReaderTest.java
+++ b/src/test/java/net/schmizz/sshj/sftp/PacketReaderTest.java
@@ -9,6 +9,7 @@ import java.io.PipedInputStream;
 import java.io.PipedOutputStream;
 import java.util.Arrays;
 
+import net.schmizz.sshj.common.SSHException;
 import net.schmizz.sshj.connection.channel.direct.Session.Subsystem;
 
 import org.junit.Before;
@@ -57,7 +58,7 @@ public class PacketReaderTest {
         try {
             reader.readPacket();
             fail("Should have failed to read packet of size " + Integer.MAX_VALUE);
-        } catch (IllegalStateException e) {
+        } catch (SSHException e) {
             e.printStackTrace();
             // success; indicated packet size was too large
         }