Updated test for writer