diff --git a/zimaos-fix.sh b/zimaos-fix.sh
index ed2792a..c8c5de5 100644
--- a/zimaos-fix.sh
+++ b/zimaos-fix.sh
@@ -5,6 +5,8 @@ temp_file=$(mktemp)
 tmpRoot="/tmp/root"
 mkdir -p "$tmpRoot"
 rootfs=$(mount | grep ' / ' | cut -d ' ' -f 1)
+p8=$(mount | grep ' /DATA ' | cut -d ' ' -f 1)
+
 mount "$rootfs" /tmp/root > /dev/null 2>&1
 
 find /mnt/overlay/etc/ -type f -exec sh -c 'for f; do [ ! -s "$f" ] && echo "$f"; done' sh {} + > "$temp_file"
@@ -20,6 +22,17 @@ done < "$temp_file"
 rm "$temp_file"
 umount "$tmpRoot"
 
+if [ -n "$p8" ]; then
+    p8_size=$(df -BG | grep "$p8" | awk '{print $2}')
+    p8_size=${p8_size%G}
+    if [ "$p8_size" -lt 2 ]; then
+        echo "Expanding..."
+        resize2fs "$p8"
+        echo "Partition 8 expanded."
+    fi
+fi
+
+
 if [ $counter -gt 0 ]; then
     echo "Done! Please reboot your device."
     exit 0