fix for failing screenshot tests