Fix screenshot builds