diff --git a/datastructures-csv/src/test/java/org/xbib/datastructures/csv/test/GeneratorTest.java b/datastructures-csv/src/test/java/org/xbib/datastructures/csv/test/GeneratorTest.java
index 9693be5..d04f68b 100644
--- a/datastructures-csv/src/test/java/org/xbib/datastructures/csv/test/GeneratorTest.java
+++ b/datastructures-csv/src/test/java/org/xbib/datastructures/csv/test/GeneratorTest.java
@@ -39,13 +39,14 @@ public class GeneratorTest {
         StringWriter writer = new StringWriter();
         Generator gen = new Generator(writer);
         gen.keys(Arrays.asList("a", "b", "c"));
+        gen.writeKeys();
         for (int i = 0; i < 1; i++) {
             gen.write("val" + i);
             gen.write("\"Hello, World\"");
             gen.write("hey look a line seperator \n");
         }
         gen.close();
-        assertEquals("val0,\"\"\"Hello, World\"\"\",\"hey look a line seperator \n\"\n", writer.toString());
+        assertEquals("a,b,c,\nval0,\"\"\"Hello, World\"\"\",\"hey look a line seperator \n\"\n", writer.toString());
     }
 
     @Test