diff --git a/src/test/java/jota/IotaCoreApiTest.java b/src/test/java/jota/IotaCoreApiTest.java
index 36d8f44..da5ba11 100644
--- a/src/test/java/jota/IotaCoreApiTest.java
+++ b/src/test/java/jota/IotaCoreApiTest.java
@@ -51,13 +51,13 @@ public class IotaCoreApiTest {
         assertThat(neighbors.getNeighbors(), IsNull.notNullValue());
     }
 
-    @Test
+    @Test(expected = IllegalAccessError.class)
     public void shouldAddNeighbors() {
         AddNeighborsResponse res = proxy.addNeighbors("udp://8.8.8.8:14265");
         assertThat(res, IsNull.notNullValue());
     }
 
-    @Test
+    @Test(expected = IllegalAccessError.class)
     public void shouldRemoveNeighbors() {
         RemoveNeighborsResponse res = proxy.removeNeighbors("udp://8.8.8.8:14265");
         assertThat(res, IsNull.notNullValue());
@@ -100,15 +100,15 @@ public class IotaCoreApiTest {
 
     }
 
-    @Test
-    public void shouldGetInclusionStates() {
+    @Test(expected = IllegalAccessError.class)
+    public void shouldNotGetInclusionStates() {
         GetInclusionStateResponse res = proxy.getInclusionStates(new String[]{"DBPECSH9YLSSTQDGERUHJBBJTKVUDBMTJLG9WPHBINGHIFOSJMDJLARTVOXXWEFQJLLBINOHCZGYFSMUEXWPPMTOFW"}, new String[]{"EJDQOQHMLJGBMFWB9WJSPRCYIGNPO9WRHDCEQXIMPVPIJ9JV9RJGVHNX9EPGXFOOKBABCVMMAAX999999"});
         assertThat(res.getStates(), IsNull.notNullValue());
     }
 
-    @Test(expected = IllegalAccessError.class)
-    public void shouldNotGetInclusionStates() {
-        GetInclusionStateResponse res = proxy.getInclusionStates(new String[]{TEST_ADDRESS_WITH_CHECKSUM}, new String[]{"DNSBRJWNOVUCQPILOQIFDKBFJMVOTGHLIMLLRXOHFTJZGRHJUEDAOWXQRYGDI9KHYFGYDWQJZKX999999"});
+    @Test
+    public void shouldGetInclusionStates() {
+        GetInclusionStateResponse res = proxy.getInclusionStates(new String[]{TEST_ADDRESS_WITH_CHECKSUM}, new String[]{proxy.getNodeInfo().getLatestSolidSubtangleMilestone()});
         assertThat(res.getStates(), IsNull.notNullValue());
     }
 
@@ -133,5 +133,6 @@ public class IotaCoreApiTest {
         assertThat(res.getBalances(), IsNull.notNullValue());
         assertThat(res.getMilestone(), IsNull.notNullValue());
         assertThat(res.getMilestoneIndex(), IsNull.notNullValue());
+        assertThat(res.getDuration(), IsNull.notNullValue());
     }
 }