Fix to screenshot