diff --git a/code/DMSDocument.php b/code/DMSDocument.php
index 4c823cb..b0b2c4c 100755
--- a/code/DMSDocument.php
+++ b/code/DMSDocument.php
@@ -804,7 +804,7 @@ class DMSDocument extends DataObject implements DMSDocumentInterface {
 
 }
 
-class DMSDocument_Controller extends Controller {
+class DMSDocument_Controller extends ContentController {
 
 	static $testMode = false;   //mode to switch for testing. Does not return document download, just document URL