Fix screenshot

This is still failing for me though.