Allow array_key_exists as a where condition