Screenshot fix
Screenshot fixes