Fixed the test case

git-svn-id: https://svn.apache.org/repos/asf/db/ddlutils/trunk@711327 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/src/test/java/org/apache/ddlutils/task/TestWriteDataToDatabaseCommand.java b/src/test/java/org/apache/ddlutils/task/TestWriteDataToDatabaseCommand.java
index 3cf34ca..26f5ab1 100644
--- a/src/test/java/org/apache/ddlutils/task/TestWriteDataToDatabaseCommand.java
+++ b/src/test/java/org/apache/ddlutils/task/TestWriteDataToDatabaseCommand.java
@@ -99,7 +99,7 @@
 
         runTask(getDatabaseToDdlTaskInstance(), dataXml, false, false);
 
-        List beans = getRows("roundtrip");
+        List beans = getRows("roundtrip", "pk");
 
         assertEquals(3, beans.size());
         assertEquals((Object)"val1", beans.get(0), "pk");
@@ -139,7 +139,7 @@
 
         runTask(getDatabaseToDdlTaskInstance(), dataXml.toString(), true, false);
 
-        List beans = getRows("roundtrip");
+        List beans = getRows("roundtrip", "avalue");
 
         assertEquals(numObjs, beans.size());
         for (int idx = 0; idx < numObjs; idx++)