Add FLATTEN fix.