diff --git a/deeplearning4j/pom.xml b/deeplearning4j/pom.xml
index 64d365e188..9a381c6967 100644
--- a/deeplearning4j/pom.xml
+++ b/deeplearning4j/pom.xml
@@ -14,6 +14,11 @@
     </parent>
 
     <dependencies>
+        <dependency>
+            <groupId>org.nd4j</groupId>
+            <artifactId>nd4j-api</artifactId>
+            <version>${dl4j.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.nd4j</groupId>
             <artifactId>nd4j-native-platform</artifactId>
@@ -43,7 +48,7 @@
     </dependencies>
 
     <properties>
-        <dl4j.version>0.9.1</dl4j.version>
+        <dl4j.version>1.0.0-beta5</dl4j.version>
     </properties>
 
 </project>
diff --git a/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java b/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java
index 1246de973f..e320c366bf 100644
--- a/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java
+++ b/deeplearning4j/src/main/java/com/baeldung/logreg/MnistClassifier.java
@@ -165,4 +165,4 @@ public class MnistClassifier {
         ModelSerializer.writeModel(model, modelPath, true);
         logger.info("The MINIST model has been saved in {}", modelPath.getPath());
     }
-}
\ No newline at end of file
+}