o First badge: UNBREAKABLE.

1 files changed, 4 insertions, 6 deletions

diff --git a/src/main/java/io/trygvis/esper/testing/core/db/BuildDto.java b/src/main/java/io/trygvis/esper/testing/core/db/BuildDto.java
index 5cc6b0a..305c93a 100644
--- a/src/main/java/io/trygvis/esper/testing/core/db/BuildDto.java
+++ b/src/main/java/io/trygvis/esper/testing/core/db/BuildDto.java
@@ -5,17 +5,15 @@ import org.joda.time.*;
 
 import java.util.*;
 
-class BuildDto extends AbstractEntity {
+public class BuildDto extends AbstractEntity {
     public final DateTime timestamp;
     public final boolean success;
-    public final UUID referenceUuid;
-    public final String referenceType;
+    public final EntityRef ref;
 
-    BuildDto(UUID uuid, DateTime createdDate, DateTime timestamp, boolean success, UUID referenceUuid, String referenceType) {
+    public BuildDto(UUID uuid, DateTime createdDate, DateTime timestamp, boolean success, EntityRef ref) {
         super(uuid, createdDate);
         this.timestamp = timestamp;
         this.success = success;
-        this.referenceUuid = referenceUuid;
-        this.referenceType = referenceType;
+        this.ref = ref;
     }
 }