diff --git a/src/test/com/cloudera/sqoop/TestIncrementalImport.java b/src/test/com/cloudera/sqoop/TestIncrementalImport.java
index fd945529..b456ca6a 100644
--- a/src/test/com/cloudera/sqoop/TestIncrementalImport.java
+++ b/src/test/com/cloudera/sqoop/TestIncrementalImport.java
@@ -547,6 +547,8 @@ private List<String> getArgListForQuery(String query, String directoryName,
     args.add(SOURCE_DB_URL);
     args.add("--query");
     args.add(query);
+    args.add("--class-name");
+    args.add(directoryName);
     args.add("--target-dir");
     args.add(BaseSqoopTestCase.LOCAL_WAREHOUSE_DIR
       + System.getProperty("file.separator") + directoryName);