diff --git a/_config/routes.yml b/_config/routes.yml
index c3a5c2f..d27a5ae 100644
--- a/_config/routes.yml
+++ b/_config/routes.yml
@@ -3,4 +3,4 @@ Name: widgetsroutes
 ---
 SilverStripe\Control\Director:
   rules:
-    'WidgetController//$Action/$ID/$OtherID': 'SilverStripe\Widgets\Controllers\WidgetController'
+    'WidgetController//$Action/$ID/$OtherID': 'SilverStripe\Widgets\Model\WidgetController'
diff --git a/src/Model/Widget.php b/src/Model/Widget.php
index 51d0c9d..27609b3 100644
--- a/src/Model/Widget.php
+++ b/src/Model/Widget.php
@@ -70,7 +70,7 @@ class Widget extends DataObject
     private static $table_name = 'Widget';
 
     private static $extensions = [
-        Versioned::class . '.versioned',
+        Versioned::class,
     ];
 
     /**
diff --git a/src/Model/WidgetArea.php b/src/Model/WidgetArea.php
index fe05577..2007ccf 100644
--- a/src/Model/WidgetArea.php
+++ b/src/Model/WidgetArea.php
@@ -27,7 +27,7 @@ class WidgetArea extends DataObject
     ];
 
     private static $extensions = [
-        Versioned::class . '.versioned',
+        Versioned::class,
     ];
 
     private static $table_name = 'WidgetArea';