diff --git a/tests/model/VirtualPageTest.php b/tests/model/VirtualPageTest.php
index 5f3591ab..5b96bf5b 100644
--- a/tests/model/VirtualPageTest.php
+++ b/tests/model/VirtualPageTest.php
@@ -272,6 +272,12 @@ class VirtualPageTest extends SapphireTest {
 		
 		// Unpublish the source page, confirm that the virtual page has also been unpublished
 		$p->doUnpublish();
+
+        // The draft VP still has the CopyContentFromID link
+		$vp->flushCache();
+		$vp = DataObject::get_by_id('SiteTree', $vp->ID);
+		$this->assertEquals($p->ID, $vp->CopyContentFromID);
+
 		$vpLive = Versioned::get_one_by_stage('SiteTree', 'Live', '"SiteTree"."ID" = ' . $vp->ID);
 		$this->assertNull($vpLive);