Dear Andrei,
It is not misaligned actually the gallery section is a masonry type. The layout would result as it is in your current site if you use different dimensions image. So you have to use equal dimension image in all gallery item to make it aligned and look like its demo.
Best Regards!
