Fix expression flattener.