diff --git a/src/samples/Sample17/CreateDocumentWorkflow.cs b/src/samples/Sample17/CreateDocumentWorkflow.cs
index 7a9e5bb42..7c7886078 100644
--- a/src/samples/Sample17/CreateDocumentWorkflow.cs
+++ b/src/samples/Sample17/CreateDocumentWorkflow.cs
@@ -1,8 +1,10 @@
 using Elsa.Activities.Console.Activities;
+using Elsa.Activities.Primitives;
 using Elsa.Expressions;
 using Elsa.Services;
 using Elsa.Services.Models;
 using Sample17.Activities;
+using Sample17.Models;
 
 namespace Sample17
 {
@@ -14,7 +16,12 @@ namespace Sample17
                 .StartWith<WriteLine>(x => x.TextExpression = new LiteralExpression("Enter a title"))
                 .Then<ReadLine>(id: "DocumentTitleInput")
                 .Then<CreateDocument>(x => x.TitleExpression = new JavaScriptExpression<string>("DocumentTitleInput.Input"), id: "CreateDocument")
-                .Then<WriteLine>(x => x.TextExpression = new JavaScriptExpression<string>("`A new document was created with title \"${CreateDocument.Document.Title}\"`"));
+                .Then<SetVariable>(x =>
+                {
+                    x.VariableName = "Document";
+                    x.ValueExpression = new JavaScriptExpression<Document>("CreateDocument.Document");
+                })
+                .Then<WriteLine>(x => x.TextExpression = new JavaScriptExpression<string>("`A new document was created with title \"${Document.Title}\"`"));
         }
     }
 }
\ No newline at end of file