Add test for Supply.flat