diff --git a/core-java-io/src/main/java/com/baeldung/zip/ZipDirectory.java b/core-java-io/src/main/java/com/baeldung/zip/ZipDirectory.java
index 7da71a093d..42147b07db 100644
--- a/core-java-io/src/main/java/com/baeldung/zip/ZipDirectory.java
+++ b/core-java-io/src/main/java/com/baeldung/zip/ZipDirectory.java
@@ -24,6 +24,13 @@ public class ZipDirectory {
             return;
         }
         if (fileToZip.isDirectory()) {
+            if (fileName.endsWith("/")) {
+                zipOut.putNextEntry(new ZipEntry(fileName));
+                zipOut.closeEntry();
+            } else {
+                zipOut.putNextEntry(new ZipEntry(fileName + "/"));
+                zipOut.closeEntry();
+            }
             final File[] children = fileToZip.listFiles();
             for (final File childFile : children) {
                 zipFile(childFile, fileName + "/" + childFile.getName(), zipOut);