Added tests for `flat` scheme