diff --git a/code/DMS.php b/code/DMS.php
index 0785ff1..966cef0 100644
--- a/code/DMS.php
+++ b/code/DMS.php
@@ -151,7 +151,7 @@ class DMS extends SS_Object implements DMSInterface
 
     public function getDocumentSetsByPage(SiteTree $page)
     {
-        return $page->DocumentSets();
+        return $page->getDocumentSets();
     }
 
     /**
diff --git a/code/extensions/DMSSiteTreeExtension.php b/code/extensions/DMSSiteTreeExtension.php
index a6f6959..abeb7de 100644
--- a/code/extensions/DMSSiteTreeExtension.php
+++ b/code/extensions/DMSSiteTreeExtension.php
@@ -55,14 +55,11 @@ class DMSSiteTreeExtension extends DataExtension
     /**
      * Get a list of document sets for the owner page
      *
-     * @deprecated 3.0 Use DocumentSets() instead.
-     *
      * @return ArrayList
      */
     public function getDocumentSets()
     {
-        Deprecation::notice('3.0', 'Use DocumentSets() instead');
-        return $this->owner->hasManyComponent('DocumentSets');
+        return $this->owner->DocumentSets();
     }
 
     /**
@@ -74,7 +71,7 @@ class DMSSiteTreeExtension extends DataExtension
     {
         $documents = ArrayList::create();
 
-        foreach ($this->owner->DocumentSets() as $documentSet) {
+        foreach ($this->getDocumentSets() as $documentSet) {
             /** @var DocumentSet $documentSet */
             $documents->merge($documentSet->getDocuments());
         }
diff --git a/code/model/DMSDocument.php b/code/model/DMSDocument.php
index 7176eab..228983f 100644
--- a/code/model/DMSDocument.php
+++ b/code/model/DMSDocument.php
@@ -820,12 +820,12 @@ class DMSDocument extends DataObject implements DMSDocumentInterface
                 new GridFieldDataColumns(),
                 new GridFieldPaginator(30)
             );
+
             $versionsGridFieldConfig->getComponentByType('GridFieldDataColumns')
                 ->setDisplayFields(Config::inst()->get('DMSDocument_versions', 'display_fields'))
                 ->setFieldFormatting(
                     array(
-                        'FilenameWithoutID' => '<a target="_blank" class="file-url" href="$Link">'
-                            . '$FilenameWithoutID</a>'
+                        'FilenameWithoutID' => '<a target=\"_blank\" class=\"file-url\" href=\"$Link\">$FilenameWithoutID</a>'
                     )
                 );
 
diff --git a/composer.json b/composer.json
index ca61e2e..1494360 100644
--- a/composer.json
+++ b/composer.json
@@ -12,7 +12,8 @@
         "silverstripe/framework": "^3.5",
         "silverstripe/cms": "^3.5",
         "symbiote/silverstripe-gridfieldextensions": "^2.0",
-        "silverstripe/taxonomy": "^1.2"
+        "silverstripe/taxonomy": "^1.2",
+        "mustangostang/spyc": "<=0.6.2"
     },
     "suggest": {
         "undefinedoffset/sortablegridfield": "Allow documents to be reordered via drag-and-drop"
diff --git a/templates/Includes/DocumentSets.ss b/templates/Includes/DocumentSets.ss
index 3998fa8..632fb29 100644
--- a/templates/Includes/DocumentSets.ss
+++ b/templates/Includes/DocumentSets.ss
@@ -1,6 +1,6 @@
-<% if $DocumentSets %>
+<% if $getDocumentSets %>
     <div class="documentsets">
-        <% loop $DocumentSets %>
+        <% loop $getDocumentSets %>
             <% include DocumentSet %>
         <% end_loop %>
     </div>
diff --git a/tests/DMSDocumentSetTest.php b/tests/DMSDocumentSetTest.php
index f1bdac4..c088618 100644
--- a/tests/DMSDocumentSetTest.php
+++ b/tests/DMSDocumentSetTest.php
@@ -50,9 +50,9 @@ class DMSDocumentSetTest extends SapphireTest
         $ds2 = $this->objFromFixture('DMSDocumentSet', 'ds2');
         $ds3 = $this->objFromFixture('DMSDocumentSet', 'ds3');
 
-        $this->assertCount(0, $s4->DocumentSets(), 'Page 4 has no document sets associated');
-        $this->assertCount(2, $s1->DocumentSets(), 'Page 1 has 2 document sets');
-        $this->assertEquals(array($ds1->ID, $ds2->ID), $s1->DocumentSets()->column('ID'));
+        $this->assertCount(0, $s4->getDocumentSets(), 'Page 4 has no document sets associated');
+        $this->assertCount(2, $s1->getDocumentSets(), 'Page 1 has 2 document sets');
+        $this->assertEquals(array($ds1->ID, $ds2->ID), $s1->getDocumentSets()->column('ID'));
     }
 
     /**
diff --git a/tests/DMSEmbargoTest.php b/tests/DMSEmbargoTest.php
index 83cda5f..484444f 100644
--- a/tests/DMSEmbargoTest.php
+++ b/tests/DMSEmbargoTest.php
@@ -142,7 +142,7 @@ class DMSEmbargoTest extends SapphireTest
         $doc->Folder = "0";
         $dID = $doc->write();
 
-        $s1->DocumentSets()->first()->getDocuments()->add($doc);
+        $s1->getDocumentSets()->first()->getDocuments()->add($doc);
 
         $s1->publish('Stage', 'Live');
         $s1->doPublish();
diff --git a/tests/tasks/MigrateToDocumentSetsTaskTest.php b/tests/tasks/MigrateToDocumentSetsTaskTest.php
index d58ea16..de26945 100644
--- a/tests/tasks/MigrateToDocumentSetsTaskTest.php
+++ b/tests/tasks/MigrateToDocumentSetsTaskTest.php
@@ -65,9 +65,9 @@ class MigrateToDocumentSetsTaskTest extends SapphireTest
         $this->assertContains('Skipped: already has a set: 1', $result);
 
         // Test that some of the relationship records were written correctly
-        $this->assertCount(1, $firstPageSets = $this->objFromFixture('SiteTree', 'one')->DocumentSets());
+        $this->assertCount(1, $firstPageSets = $this->objFromFixture('SiteTree', 'one')->getDocumentSets());
         $this->assertSame('Default', $firstPageSets->first()->Title);
-        $this->assertCount(1, $this->objFromFixture('SiteTree', 'two')->DocumentSets());
+        $this->assertCount(1, $this->objFromFixture('SiteTree', 'two')->getDocumentSets());
 
         // With dryrun enabled and being run the second time, nothing should be done
         $result = $this->runTask(array('action' => 'create-default-document-set', 'dryrun' => '1'));