Fixed a bug with the gallery