Maintain Shopify Collage Format On Mobile

[KhallP]

I’m trying to keep the formatting of my desktop collage on mobile but when I switch the view to mobile the formatting is completely disconfigured

I inserted the following code in collage.css to keep the collage images the same size:

@media (min-width: 310px) {
    .collage__item--left:nth-child(3n - 2) {
        grid-column: 1 !important;
        grid-row: 1 !important;
    }
    .collage__item--left:nth-child(3n - 1), 
    .collage__item--left:nth-child(3n) {
        grid-column-start: unset !important;
    }

}

Desktop Aspect:

Mobile Aspect:

If possible, I want to keep the mobile look the same as the desktop