diff --git a/model/DataQuery.php b/model/DataQuery.php
index fa3d8b1dc..2445df599 100644
--- a/model/DataQuery.php
+++ b/model/DataQuery.php
@@ -227,7 +227,7 @@ class DataQuery {
 
 		// TODO: Versioned, Translatable, SiteTreeSubsites, etc, could probably be better implemented as subclasses of DataQuery
 
-		$obj = Injector::inst()->get(ClassInfo::baseDataClass($this->dataClass));
+		$obj = Injector::inst()->get($this->dataClass);
 		$obj->extend('augmentSQL', $query, $this);
 
 		$this->ensureSelectContainsOrderbyColumns($query);