diff --git a/_config.php b/_config.php
index ccab746..82eba82 100644
--- a/_config.php
+++ b/_config.php
@@ -1,5 +1,3 @@
 <?php
 
-if(class_exists('WidgetArea')) DataExtension::add_to_class('BlogTree','BlogTreeExtension');
-
-?>
+if(class_exists('WidgetArea')) Object::add_extension('BlogTree','BlogTreeExtension');
\ No newline at end of file