Fixed a bug with the screenshots