diff --git a/src/java/com/cloudera/sqoop/mapreduce/HBaseImportJob.java b/src/java/com/cloudera/sqoop/mapreduce/HBaseImportJob.java
index 7767ddc3..07bc215d 100644
--- a/src/java/com/cloudera/sqoop/mapreduce/HBaseImportJob.java
+++ b/src/java/com/cloudera/sqoop/mapreduce/HBaseImportJob.java
@@ -35,7 +35,6 @@
 import org.apache.hadoop.mapreduce.Job;
 import org.apache.hadoop.mapreduce.Mapper;
 import org.apache.hadoop.mapreduce.OutputFormat;
-import com.cloudera.sqoop.mapreduce.db.DataDrivenDBInputFormat;
 
 import com.cloudera.sqoop.SqoopOptions;
 import com.cloudera.sqoop.hbase.HBasePutProcessor;
@@ -56,7 +55,7 @@ public class HBaseImportJob extends DataDrivenImportJob {
 
   public HBaseImportJob(final SqoopOptions opts,
       final ImportJobContext importContext) {
-    super(opts, DataDrivenDBInputFormat.class, importContext);
+    super(opts, importContext.getInputFormat(), importContext);
   }
 
   @Override