Fix screenshot output