diff --git a/cdap-app-templates/cdap-etl/hydrator-test/src/main/java/io/cdap/cdap/etl/mock/test/HydratorTestBase.java b/cdap-app-templates/cdap-etl/hydrator-test/src/main/java/io/cdap/cdap/etl/mock/test/HydratorTestBase.java
index b7625bb97f23..90f72138c133 100644
--- a/cdap-app-templates/cdap-etl/hydrator-test/src/main/java/io/cdap/cdap/etl/mock/test/HydratorTestBase.java
+++ b/cdap-app-templates/cdap-etl/hydrator-test/src/main/java/io/cdap/cdap/etl/mock/test/HydratorTestBase.java
@@ -33,6 +33,7 @@
 import io.cdap.cdap.etl.api.engine.sql.capability.PushCapability;
 import io.cdap.cdap.etl.api.engine.sql.dataset.SQLDataset;
 import io.cdap.cdap.etl.api.engine.sql.request.SQLPushRequest;
+import io.cdap.cdap.etl.api.exception.ErrorDetailsProviderSpec;
 import io.cdap.cdap.etl.api.join.AutoJoiner;
 import io.cdap.cdap.etl.api.join.error.JoinError;
 import io.cdap.cdap.etl.api.lineage.AccessType;
@@ -177,6 +178,7 @@ protected static void setupBatchArtifacts(ArtifactId artifactId, Class<?> appCla
         PushCapability.class.getPackage().getName(),
         SparkRecordCollection.class.getPackage().getName(),
         Connector.class.getPackage().getName(),
+        ErrorDetailsProviderSpec.class.getPackage().getName(),
         "org.apache.avro.mapred", "org.apache.avro", "org.apache.avro.generic",
         "org.apache.avro.io");
 
@@ -210,6 +212,7 @@ protected static void setupStreamingArtifacts(ArtifactId artifactId, Class<?> ap
         SparkCompute.class.getPackage().getName(),
         InvalidStageException.class.getPackage().getName(),
         PipelineConfigurable.class.getPackage().getName(),
+        ErrorDetailsProviderSpec.class.getPackage().getName(),
         // have to export this package, otherwise getting ClassCastException in unit test
         FieldOperation.class.getPackage().getName());