Screenshot fix bug