Screenshot fix