diff --git a/core/src/test/scala/unit/kafka/network/SocketServerTest.scala b/core/src/test/scala/unit/kafka/network/SocketServerTest.scala
index c3b1ac4..53e11d5 100644
--- a/core/src/test/scala/unit/kafka/network/SocketServerTest.scala
+++ b/core/src/test/scala/unit/kafka/network/SocketServerTest.scala
@@ -125,7 +125,7 @@ class SocketServerTest extends JUnitSuite {
         5000))
   }
 
-  @Test(expected = classOf[SocketException])
+  @Test(expected = classOf[IOException])
   def testSocketsCloseOnShutdown() {
     // open a connection and then shutdown the server
     val socket = connect()
@@ -135,3 +135,4 @@ class SocketServerTest extends JUnitSuite {
     sendRequest(socket, 0, bytes)
   }
 }
+