diff --git a/graylog2-server/src/test/java/org/graylog2/database/PaginatedDbServiceTest.java b/graylog2-server/src/test/java/org/graylog2/database/PaginatedDbServiceTest.java
index 039c6fd11b4a..a77442b36508 100644
--- a/graylog2-server/src/test/java/org/graylog2/database/PaginatedDbServiceTest.java
+++ b/graylog2-server/src/test/java/org/graylog2/database/PaginatedDbServiceTest.java
@@ -26,7 +26,6 @@
 import org.graylog2.bindings.providers.MongoJackObjectMapperProvider;
 import org.junit.After;
 import org.junit.Before;
-import org.junit.ClassRule;
 import org.junit.Rule;
 import org.junit.Test;
 import org.mongojack.DBQuery;
@@ -56,7 +55,7 @@ public static class TestDTO {
         public String title;
 
         @JsonCreator
-        public TestDTO(@JsonProperty("id") String id, @JsonProperty("title") String title) {
+        public TestDTO(@JsonProperty("id") @Id String id, @JsonProperty("title") String title) {
             this.id = id;
             this.title = title;
         }
diff --git a/graylog2-server/src/test/java/org/graylog2/security/encryption/EncryptedValueTest.java b/graylog2-server/src/test/java/org/graylog2/security/encryption/EncryptedValueTest.java
index 543956131536..6227846fe69c 100644
--- a/graylog2-server/src/test/java/org/graylog2/security/encryption/EncryptedValueTest.java
+++ b/graylog2-server/src/test/java/org/graylog2/security/encryption/EncryptedValueTest.java
@@ -203,7 +203,7 @@ public static TestDTO create(@JsonProperty EncryptedValue passwordValue) {
         }
 
         @JsonCreator
-        public static TestDTO create(@JsonProperty("id") String id, @JsonProperty("password_value") EncryptedValue passwordValue) {
+        public static TestDTO create(@JsonProperty("id") @Id String id, @JsonProperty("password_value") EncryptedValue passwordValue) {
             return new AutoValue_EncryptedValueTest_TestDTO(id, passwordValue);
         }
     }