Flexbox Wrap not wrapping correctly

Hi guys,

I’ve designed my portfolio using CMS collections, and I have a collection list with flexbox and a wrap that should be displaying either 2 small pictures in a row or 1 big like on this preview:

Only problem is that when I publish, the small images are scaled up and fill the width, even though layout looks goo on preview.

Any help with this would be much appreciated.