Fix for broken screenshot