Fix for busted screenshot test