Fix screenshots not working