diff --git a/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/convert/UpdateMapperUnitTests.java b/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/convert/UpdateMapperUnitTests.java index 8a80fe8e8..d6942a15f 100644 --- a/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/convert/UpdateMapperUnitTests.java +++ b/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/convert/UpdateMapperUnitTests.java @@ -992,11 +992,10 @@ public class UpdateMapperUnitTests { DBObject mappedUpdate = mapper.getMappedObject(update.getUpdateObject(), context.getPersistentEntity(EntityWithObjectMap.class)); - DBObject $set = getAsDBObject(mappedUpdate, "$set"); - DBObject mapToSet = getAsDBObject($set, "map"); + DBObject mapToSet = getAsDBObject(getAsDBObject(mappedUpdate, "$set"), "map"); for (Object key : mapToSet.keySet()) { - assertThat(key, instanceOf(String.class)); + assertThat(key, is(instanceOf(String.class))); } }