The validation was comparing image height to itself instead of comparing height to width, causing the square check to always pass even for non-square images. Co-authored-by: Dhairyashil Shinde <93669429+dhairyashiil@users.noreply.github.com>