Fix screenshot checks