diff --git a/model/DataExtension.php b/model/DataExtension.php
index 041630200..795a6332c 100644
--- a/model/DataExtension.php
+++ b/model/DataExtension.php
@@ -42,7 +42,7 @@ abstract class DataExtension extends Extension {
 	 *
 	 * @param SQLQuery $query Query to augment.
 	 */
-	public function augmentSQL(SQLQuery &$query) {
+	public function augmentSQL(SQLQuery &$query, DataQuery &$dataQuery = null) {
 	}
 
 	/**
diff --git a/model/Hierarchy.php b/model/Hierarchy.php
index d4f6c667c..8309406fe 100644
--- a/model/Hierarchy.php
+++ b/model/Hierarchy.php
@@ -38,7 +38,7 @@ class Hierarchy extends DataExtension {
 	 */
 	private static $node_threshold_leaf = 250;
 	
-	public function augmentSQL(SQLQuery &$query) {
+	public function augmentSQL(SQLQuery &$query, DataQuery &$dataQuery = null) {
 	}
 
 	public function augmentDatabase() {