Test flat=True case