--- a/java/sql-dk/src/info/globalcode/sql/dk/formatting/AbstractFormatter.java	Mon Dec 23 00:14:29 2013 +0100
+++ b/java/sql-dk/src/info/globalcode/sql/dk/formatting/AbstractFormatter.java	Mon Dec 23 11:50:24 2013 +0100
@@ -170,7 +170,7 @@
 
 	@Override
 	public void writeColumnsHeader(ColumnsHeader header) {
-		peekState(EnumSet.of(State.RESULT_SET, State.UPDATES_RESULT));
+		peekState(EnumSet.of(State.RESULT_SET, State.GENERATED_KEYS));
 
 		if (currentColumnsHeader == null) {
 			currentColumnsHeader = header;
@@ -204,7 +204,7 @@
 
 	@Override
 	public void writeStartUpdatesResult() {
-		pushState(State.RESULT_SET, EnumSet.of(State.DATABASE));
+		pushState(State.UPDATES_RESULT, EnumSet.of(State.DATABASE));
 	}
 
 	@Override