diff --git a/datavec/datavec-data/datavec-data-image/pom.xml b/datavec/datavec-data/datavec-data-image/pom.xml
index aef66381ac1b..3cdc8afa6d50 100644
--- a/datavec/datavec-data/datavec-data-image/pom.xml
+++ b/datavec/datavec-data/datavec-data-image/pom.xml
@@ -117,6 +117,13 @@
                     <classpathDependencyExcludes>
                         <classpathDependencyExclude>com.google.android:android</classpathDependencyExclude>
                     </classpathDependencyExcludes>
+                    <includes>
+                        <include>*.java</include>
+                        <include>**/*.java</include>
+                        <include>**/Test*.java</include>
+                        <include>**/*Test.java</include>
+                        <include>**/*TestCase.java</include>
+                    </includes>
                 </configuration>
             </plugin>
         </plugins>
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/LabelGeneratorTest.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/LabelGeneratorTest.java
index 44c32f94fd7c..4c234947cb1c 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/LabelGeneratorTest.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/LabelGeneratorTest.java
@@ -24,6 +24,7 @@
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
 import org.nd4j.common.io.ClassPathResource;
+import org.nd4j.common.tests.BaseND4JTest;
 
 import java.io.File;
 import java.util.Arrays;
@@ -32,7 +33,7 @@
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
 
-public class LabelGeneratorTest {
+public class LabelGeneratorTest extends BaseND4JTest {
 
     @Rule
     public TemporaryFolder testDir = new TemporaryFolder();
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/LoaderTests.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/LoaderTests.java
index 7afdb7ac0ff0..d21ec2132abe 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/LoaderTests.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/LoaderTests.java
@@ -16,34 +16,25 @@
 
 package org.datavec.image.loader;
 
-import org.apache.commons.io.FileUtils;
 import org.apache.commons.io.FilenameUtils;
-import org.datavec.api.io.filters.BalancedPathFilter;
 import org.datavec.api.records.reader.RecordReader;
-import org.datavec.api.split.FileSplit;
-import org.datavec.api.split.InputSplit;
-import org.datavec.image.recordreader.ImageRecordReader;
 import org.junit.Ignore;
 import org.junit.Test;
+import org.nd4j.common.tests.BaseND4JTest;
 import org.nd4j.linalg.dataset.DataSet;
 
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.InputStream;
-import java.io.SequenceInputStream;
-import java.util.Collection;
-import java.util.Iterator;
 import java.util.List;
 import java.util.Random;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.*;
 
 /**
  *
  */
-public class LoaderTests {
+public class LoaderTests extends BaseND4JTest {
 
     private static void ensureDataAvailable(){
         //Ensure test resources available by initializing CifarLoader and relying on auto download
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestImageLoader.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestImageLoader.java
index 1683980f0acb..f1336270e7f6 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestImageLoader.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestImageLoader.java
@@ -17,6 +17,7 @@
 package org.datavec.image.loader;
 
 import org.junit.Test;
+import org.nd4j.common.tests.BaseND4JTest;
 import org.nd4j.linalg.api.ndarray.INDArray;
 
 import java.awt.image.BufferedImage;
@@ -25,7 +26,7 @@
 import static org.junit.Assert.assertEquals;
 
 
-public class TestImageLoader {
+public class TestImageLoader extends BaseND4JTest {
 
     private static long seed = 10;
     private static Random rng = new Random(seed);
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestNativeImageLoader.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestNativeImageLoader.java
index 6e77055699d5..3a30236a64d8 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestNativeImageLoader.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/loader/TestNativeImageLoader.java
@@ -28,6 +28,7 @@
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
+import org.nd4j.common.tests.BaseND4JTest;
 import org.nd4j.linalg.api.buffer.DataType;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.factory.Nd4j;
@@ -55,7 +56,7 @@
  * @author saudet
  */
 @Slf4j
-public class TestNativeImageLoader {
+public class TestNativeImageLoader extends BaseND4JTest {
     static final long seed = 10;
     static final Random rng = new Random(seed);
 
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/FileBatchRecordReaderTest.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/FileBatchRecordReaderTest.java
index 75020ebf1be4..3b2e2f1d20b5 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/FileBatchRecordReaderTest.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/FileBatchRecordReaderTest.java
@@ -28,6 +28,7 @@
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
 import org.nd4j.common.loader.FileBatch;
+import org.nd4j.common.tests.BaseND4JTest;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.common.io.ClassPathResource;
 
@@ -36,7 +37,7 @@
 
 import static org.junit.Assert.*;
 
-public class FileBatchRecordReaderTest {
+public class FileBatchRecordReaderTest extends BaseND4JTest {
 
     @Rule
     public TemporaryFolder testDir = new TemporaryFolder();
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestImageRecordReader.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestImageRecordReader.java
index fdcbe959a627..e98c45da80e3 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestImageRecordReader.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestImageRecordReader.java
@@ -36,6 +36,7 @@
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
 import org.nd4j.common.resources.Resources;
+import org.nd4j.common.tests.BaseND4JTest;
 import org.nd4j.linalg.api.buffer.DataType;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.factory.Nd4j;
@@ -53,7 +54,7 @@
 /**
  * Created by Alex on 27/09/2016.
  */
-public class TestImageRecordReader {
+public class TestImageRecordReader extends BaseND4JTest {
 
     @Rule
     public TemporaryFolder testDir = new TemporaryFolder();
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestObjectDetectionRecordReader.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestObjectDetectionRecordReader.java
index d8620096abde..1b9f9f0e1e08 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestObjectDetectionRecordReader.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/TestObjectDetectionRecordReader.java
@@ -34,6 +34,7 @@
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
+import org.nd4j.common.tests.BaseND4JTest;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.factory.Nd4j;
 import org.nd4j.linalg.indexing.BooleanIndexing;
@@ -49,7 +50,7 @@
 
 import static org.junit.Assert.*;
 
-public class TestObjectDetectionRecordReader {
+public class TestObjectDetectionRecordReader extends BaseND4JTest {
 
     @Rule
     public TemporaryFolder testDir = new TemporaryFolder();
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/objdetect/TestVocLabelProvider.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/objdetect/TestVocLabelProvider.java
index 9d12ced22e57..b09e015b732f 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/objdetect/TestVocLabelProvider.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/recordreader/objdetect/TestVocLabelProvider.java
@@ -21,6 +21,7 @@
 import org.junit.Test;
 import org.junit.rules.TemporaryFolder;
 import org.nd4j.common.io.ClassPathResource;
+import org.nd4j.common.tests.BaseND4JTest;
 
 import java.io.File;
 import java.util.Arrays;
@@ -29,7 +30,7 @@
 
 import static org.junit.Assert.assertEquals;
 
-public class TestVocLabelProvider {
+public class TestVocLabelProvider extends BaseND4JTest {
 
     @Rule
     public TemporaryFolder testDir = new TemporaryFolder();
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/JsonYamlTest.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/JsonYamlTest.java
index 9825e6899885..47ac661adbfc 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/JsonYamlTest.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/JsonYamlTest.java
@@ -18,6 +18,7 @@
 
 import org.datavec.image.data.ImageWritable;
 import org.junit.Test;
+import org.nd4j.common.tests.BaseND4JTest;
 
 import java.io.IOException;
 import java.util.List;
@@ -29,7 +30,7 @@
 /**
  * Created by kepricon on 17. 5. 25.
  */
-public class JsonYamlTest {
+public class JsonYamlTest extends BaseND4JTest {
     @Test
     public void testJsonYamlImageTransformProcess() throws IOException {
         int seed = 12345;
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/ResizeImageTransformTest.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/ResizeImageTransformTest.java
index d769aac42f34..fb3661dd419e 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/ResizeImageTransformTest.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/ResizeImageTransformTest.java
@@ -20,6 +20,7 @@
 import org.datavec.image.data.ImageWritable;
 import org.junit.Before;
 import org.junit.Test;
+import org.nd4j.common.tests.BaseND4JTest;
 
 import static org.junit.Assert.assertEquals;
 
@@ -28,7 +29,7 @@
  *
  * @author raver119@gmail.com
  */
-public class ResizeImageTransformTest {
+public class ResizeImageTransformTest extends BaseND4JTest {
     @Before
     public void setUp() throws Exception {
 
diff --git a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/TestImageTransform.java b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/TestImageTransform.java
index bd71a10298ee..c01bf892bcfa 100644
--- a/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/TestImageTransform.java
+++ b/datavec/datavec-data/datavec-data-image/src/test/java/org/datavec/image/transform/TestImageTransform.java
@@ -33,6 +33,7 @@
 import java.util.Random;
 
 import org.bytedeco.opencv.opencv_core.*;
+import org.nd4j.common.tests.BaseND4JTest;
 
 import static org.bytedeco.opencv.global.opencv_core.*;
 import static org.bytedeco.opencv.global.opencv_imgproc.*;
@@ -42,7 +43,7 @@
  *
  * @author saudet
  */
-public class TestImageTransform {
+public class TestImageTransform extends BaseND4JTest {
     static final long seed = 10;
     static final Random rng = new Random(seed);
     static final OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();