diff --git a/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisClientLibLeaseCoordinatorTest.java b/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisClientLibLeaseCoordinatorTest.java
index d8549870..e48cdd16 100644
--- a/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisClientLibLeaseCoordinatorTest.java
+++ b/src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisClientLibLeaseCoordinatorTest.java
@@ -79,7 +79,7 @@ public class KinesisClientLibLeaseCoordinatorTest {
         leaseCoordinator.initialize();
     }
 
-    @Test(expected = KinesisClientLibIOException.class)
+    @Test(expected = com.amazonaws.services.kinesis.clientlibrary.exceptions.InvalidStateException.class)
     public void testGetCheckpointObjectWithNoLease()
             throws DependencyException, ProvisionedThroughputException, IllegalStateException, InvalidStateException,
             KinesisClientLibException {