diff --git a/code/DMSDocument.php b/code/DMSDocument.php
index 83eaf61..4cbd478 100755
--- a/code/DMSDocument.php
+++ b/code/DMSDocument.php
@@ -804,7 +804,7 @@ class DMSDocument extends DataObject implements DMSDocumentInterface {
 
 }
 
-class DMSDocument_Controller extends ContentController {
+class DMSDocument_Controller extends Controller {
 
 	static $testMode = false;   //mode to switch for testing. Does not return document download, just document URL
 
@@ -812,6 +812,11 @@ class DMSDocument_Controller extends ContentController {
 		'index'
 	);
 
+	public function init() {
+		Versioned::choose_site_stage();
+		parent::init();
+	}
+
 	/**
 	 * Returns the document object from the request object's ID parameter.
 	 * Returns null, if no document found