diff --git a/code/extensions/MultiFormObjectDecorator.php b/code/extensions/MultiFormObjectDecorator.php
index 6fb5052..690a8d5 100644
--- a/code/extensions/MultiFormObjectDecorator.php
+++ b/code/extensions/MultiFormObjectDecorator.php
@@ -24,7 +24,7 @@ class MultiFormObjectDecorator extends DataExtension {
 		'MultiFormSession' => 'MultiFormSession',
 	);
 	
-	public function augmentSQL(SQLQuery &$query) {
+	public function augmentSQL(SQLSelect $query) {
 		// If you're querying by ID, ignore the sub-site - this is a bit ugly...
 		if(
 			strpos($query->where[0], ".`ID` = ") === false