Fix screenshot display