Fix for screenshot