diff --git a/nd4j/samediff-import/pom.xml b/nd4j/samediff-import/pom.xml
index 836be7d04..890dc063b 100644
--- a/nd4j/samediff-import/pom.xml
+++ b/nd4j/samediff-import/pom.xml
@@ -55,7 +55,8 @@
     <maven.compiler.target>1.8</maven.compiler.target>
     <java.version>1.8</java.version>
     <maven-shade-plugin.version>3.1.1</maven-shade-plugin.version>
-
+    <test.heap.size>2g</test.heap.size>
+    <test.offheap.size>2g</test.offheap.size>
   </properties>