diff --git a/replay-tool-test/replay-tool-domain/src/main/java/uk/gov/justice/framework/tools/entity/Document.java b/replay-tool-test/replay-tool-domain/src/main/java/uk/gov/justice/framework/tools/entity/Document.java
index 54ffdd4..8d19430 100644
--- a/replay-tool-test/replay-tool-domain/src/main/java/uk/gov/justice/framework/tools/entity/Document.java
+++ b/replay-tool-test/replay-tool-domain/src/main/java/uk/gov/justice/framework/tools/entity/Document.java
@@ -1,5 +1,6 @@
 package uk.gov.justice.framework.tools.entity;
 
+import java.io.Serializable;
 import java.util.UUID;
 
 import javax.persistence.Column;
@@ -9,7 +10,7 @@
 
 @Entity
 @Table(name = "document")
-public class Document {
+public class Document implements Serializable{
 
     @Id
     @Column(name = "document_id")
diff --git a/replay-tool-test/replay-tool-it-example-listener/src/main/java/uk/gov/justice/framework/tools/listener/TestEventListener.java b/replay-tool-test/replay-tool-it-example-listener/src/main/java/uk/gov/justice/framework/tools/listener/TestEventListener.java
index e2adc93..b9aaae7 100644
--- a/replay-tool-test/replay-tool-it-example-listener/src/main/java/uk/gov/justice/framework/tools/listener/TestEventListener.java
+++ b/replay-tool-test/replay-tool-it-example-listener/src/main/java/uk/gov/justice/framework/tools/listener/TestEventListener.java
@@ -5,7 +5,6 @@
 import uk.gov.justice.framework.tools.entity.Document;
 import uk.gov.justice.framework.tools.entity.Test;
 import uk.gov.justice.framework.tools.repository.TestViewstoreRepository;
-import uk.gov.justice.services.core.annotation.Component;
 import uk.gov.justice.services.core.annotation.Handles;
 import uk.gov.justice.services.core.annotation.ServiceComponent;
 import uk.gov.justice.services.messaging.JsonEnvelope;