Improve TestSimpleWriter test