diff --git a/LeanMapper/Fluent.php b/LeanMapper/Fluent.php
index <HASH>..<HASH> 100644
--- a/LeanMapper/Fluent.php
+++ b/LeanMapper/Fluent.php
@@ -24,6 +24,15 @@ class Fluent extends DibiFluent
 {
 
 	/** @var array */
+	public static $masks = array( // fixes missing UNION in dibi
+		'SELECT' => array('SELECT', 'DISTINCT', 'FROM', 'WHERE', 'GROUP BY',
+			'HAVING', 'ORDER BY', 'LIMIT', 'OFFSET', 'UNION'),
+		'UPDATE' => array('UPDATE', 'SET', 'WHERE', 'ORDER BY', 'LIMIT'),
+		'INSERT' => array('INSERT', 'INTO', 'VALUES', 'SELECT'),
+		'DELETE' => array('DELETE', 'FROM', 'USING', 'WHERE', 'ORDER BY', 'LIMIT'),
+	);
+
+	/** @var array */
 	private $relatedKeys;
Added missing UNION in Fluent:: (TODO: send pull to dibi)